fix: fix a minor bug in DS eval (#1012)

This commit is contained in:
Yuante Li
2025-07-02 23:31:29 +08:00
committed by GitHub
parent d589b859d0
commit fb8f2b561e
8 changed files with 12 additions and 12 deletions
@@ -52,7 +52,7 @@ class FeatureCoSTEEREvaluator(CoSTEEREvaluator):
result = implementation.run(env=env, entry=f"python {fname}")
if "main.py" in implementation.file_dict and result.ret_code == 0:
if "main.py" in implementation.file_dict and result.exit_code == 0:
workflow_stdout = implementation.execute(env=env, entry="python main.py")
workflow_stdout = remove_eda_part(workflow_stdout)
else:
@@ -76,6 +76,6 @@ class FeatureCoSTEEREvaluator(CoSTEEREvaluator):
user_prompt=user_prompt,
init_kwargs_update_func=FeatureEvalFeedback.val_and_update_init_dict,
)
fb.final_decision = fb.final_decision and result.ret_code == 0
fb.final_decision = fb.final_decision and result.exit_code == 0
return fb