Update utils.py

geekan · Oct 30, 2024 · a01156d · a01156d
1 parent 6b8b9c8
commit a01156d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/metagpt/ext/aflow/scripts/utils.py b/metagpt/ext/aflow/scripts/utils.py
@@ -16,7 +16,7 @@ class CodeDataset(Enum):
 
 
 def extract_test_cases_from_jsonl(entry_point: str, dataset: CodeDataset = CodeDataset.HUMAN_EVAL):
-    if dataset == CodeDataset.HUMAN_EVAL:
+    if dataset == CodeDataset.HUMAN_EVAL.value:
         file_path = "metagpt/ext/aflow/data/humaneval_public_test.jsonl"
         # Retain the original hardcoded test cases
         hardcoded_cases = {
@@ -31,7 +31,7 @@ def extract_test_cases_from_jsonl(entry_point: str, dataset: CodeDataset = CodeD
             "sum_squares": "",
             "starts_one_ends": "",
         }
-    elif dataset == CodeDataset.MBPP:
+    elif dataset == CodeDataset.MBPP.value:
         file_path = "metagpt/ext/aflow/data/mbpp_public_test.jsonl"
         hardcoded_cases = {
             "remove_odd": "",