Merge pull request #4 from futurice/feature-math-coding

update outputs
futurice · Dec 5, 2024 · 6b6520f · 6b6520f
2 parents 167b179 + c958991
commit 6b6520f
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/math_coding/flows/math_evaluation_flow/flow.dag.yaml b/math_coding/flows/math_evaluation_flow/flow.dag.yaml
@@ -8,7 +8,7 @@ inputs:
     default: "2"
 outputs:
   score:
-    type: integer
+    type: string
     reference: ${line_process.output}
   metrics:
     type: object

diff --git a/math_coding/flows/math_evaluation_flow/line_process.py b/math_coding/flows/math_evaluation_flow/line_process.py
@@ -1,4 +1,4 @@
-from promptflow.core import tool
+from promptflow import tool
 
 
 @tool
@@ -7,18 +7,18 @@ def line_process(groundtruth: str, prediction: str):
 
     if prediction == "JSONDecodeError" or prediction.startswith("Unknown Error:"):
         processed_result = -1
-        return {"output": processed_result}
+        return {"output": str(processed_result)}
 
     try:
         groundtruth = float(groundtruth)
         prediction = float(prediction)
     except ValueError:
         processed_result = -1
-        return {"output": processed_result}
+        return {"output": str(processed_result)}
 
     if round(prediction, 2) == round(groundtruth, 2):
         processed_result = 1
-    return {"output": processed_result}
+    return {"output": str(processed_result)}
 
 
 if __name__ == "__main__":