From 4def81efbfe236ebd456747cbdbbf5701f8a2b34 Mon Sep 17 00:00:00 2001 From: goulustis Date: Mon, 19 Jan 2026 22:10:33 +0800 Subject: [PATCH] add inference time --- lang_agent/eval/evaluator.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/lang_agent/eval/evaluator.py b/lang_agent/eval/evaluator.py index 0c3c3df..b31f9c7 100644 --- 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}']