apunkt/MetaGPT
1
0
Fork 0
mirror of https://github.com/FoundationAgents/MetaGPT.git synced 2026-04-25 00:36:55 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Update utils.py

Browse source
This commit is contained in:
didi 2024-10-30 15:34:15 +08:00
parent 6b8b9c8440
commit a01156de33
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
metagpt/ext/aflow/scripts/utils.py
View file

@ -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": "",