fix: return value for model quality regression to json (#91)

radicalbit · Jul 8, 2024 · ed4ffd6 · ed4ffd6
1 parent 65f4f65
commit ed4ffd6
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/spark/jobs/current_job.py b/spark/jobs/current_job.py
@@ -112,7 +112,9 @@ def main(
             complete_record["DATA_QUALITY"] = data_quality.model_dump_json(
                 serialize_as_any=True
             )
-            complete_record["MODEL_QUALITY"] = model_quality
+            complete_record["MODEL_QUALITY"] = orjson.dumps(model_quality).decode(
+                "utf-8"
+            )
 
     schema = StructType(
         [