add validator and test

Striveworks · Aug 22, 2024 · 750d0b5 · 750d0b5
1 parent b97c9dc
commit 750d0b5
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 0 deletions.
diff --git a/api/tests/unit-tests/schemas/test_evaluation.py b/api/tests/unit-tests/schemas/test_evaluation.py
@@ -382,3 +382,20 @@ def test_EvaluationResponse():
             created_at=datetime.now(),
             meta={},
         )
+
+    # test validation for meta
+    er = schemas.EvaluationResponse(
+        id=1,
+        dataset_names=["ds"],
+        model_name="test",
+        filters=schemas.Filter(),
+        parameters=schemas.EvaluationParameters(
+            task_type=enums.TaskType.CLASSIFICATION,
+        ),
+        status=enums.EvaluationStatus.DONE,
+        metrics=[],
+        confusion_matrices=[],
+        created_at=datetime.now(),
+        meta=None,
+    )
+    assert er.meta == {}
diff --git a/api/valor_api/schemas/evaluation.py b/api/valor_api/schemas/evaluation.py
@@ -286,3 +286,9 @@ class EvaluationResponse(BaseModel):
     model_config = ConfigDict(
         extra="allow", protected_namespaces=("protected_",)
     )
+
+    # make sure that `meta` is a dictionary
+    @field_validator("meta")
+    @classmethod
+    def null_to_empty_dict(cls, v):
+        return v or {}