From a01156de3325836eaac10d96dd0c1e7f795e5a7f Mon Sep 17 00:00:00 2001
From: didi <84363704+didiforgithub@users.noreply.github.com>
Date: Wed, 30 Oct 2024 15:34:15 +0800
Subject: [PATCH] Update utils.py

---
 metagpt/ext/aflow/scripts/utils.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/metagpt/ext/aflow/scripts/utils.py b/metagpt/ext/aflow/scripts/utils.py
index c2fdd0cb7..5e6222dc4 100644
--- 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": "",