Update BaseModel output comparison in GenericLMFunctionOptimizer

zenbase-ai · Aug 1, 2024 · 80603f8 · 80603f8
1 parent 63604fe
commit 80603f8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/py/src/zenbase/predefined/generic_lm_function/optimizer.py b/py/src/zenbase/predefined/generic_lm_function/optimizer.py
@@ -90,7 +90,7 @@ def optimize(self) -> Result:
     def _create_default_evaluator(self):
         def evaluator(output: BaseModel, ideal_output: dict) -> dict:
             return {
-                "passed": int(output.dict() == ideal_output),
+                "passed": int(output.model_dump(mode="json") == ideal_output),
             }
 
         return evaluator