add inference time

2026-01-19 22:10:33 +08:00
parent c06e87f043
commit 4def81efbf
1 changed files with 1 additions and 1 deletions
--- a/lang_agent/eval/evaluator.py
+++ b/lang_agent/eval/evaluator.py
@@ -82,7 +82,7 @@ class Evaluator:
        logger.info(f"saving experiment results to: {exp_save_f}")
        df.to_csv(exp_save_f, index=False)

-        metric_col = [e for e in df.columns if "feedback" in e and not e.endswith(".comment")]
+        metric_col = [e for e in df.columns if "feedback" in e and not e.endswith(".comment")] + ["execution_time"]

        df_curr_m = df[metric_col].mean().to_frame().T
        df_curr_m.index = [f'{osp.basename(head_path)}-{n_exp}']