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

fix special instruction bug

Browse source
This commit is contained in:
Yizhou Chi 2024-10-15 16:19:02 +08:00
parent d179982949
commit 0166834ce4
1 changed files with 5 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

8
expo/data/custom_task.py
View file

@ -34,14 +34,16 @@ def get_mle_task_id(dataset_dir):
return dataset_dir.split("/")[-3]
def get_mle_bench_requirements(dataset_dir, data_config, obfuscated=False, special_instruction=""):
def get_mle_bench_requirements(dataset_dir, data_config, special_instruction, obfuscated=False):
work_dir = data_config["work_dir"]
task = get_mle_task_id(dataset_dir)
output_dir = f"{work_dir}/{task}"
final_output_dir = f"{work_dir}/submission"
os.makedirs(output_dir, exist_ok=True)
special_instruction = SPECIAL_INSTRUCTIONS[special_instruction]
if special_instruction:
special_instruction = SPECIAL_INSTRUCTIONS[special_instruction]
else:
special_instruction = ""
if obfuscated:
instructions = INSTRUCTIONS_OBFUSCATED.format(dataset_dir=dataset_dir, output_dir=final_output_dir)
task_file = "description_obfuscated.md"