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

Merge pull request #1554 from didiforgithub/main

Browse source 
Fix bug for AFLOW.
This commit is contained in:
Alexander Wu 2024-10-30 15:36:31 +08:00 • committed by GitHub
commit 2077dd5173
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
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": "",