round per task display

All-Hands-AI · Nov 28, 2024 · 2dc5f83 · 2dc5f83
1 parent 6630fe9
commit 2dc5f83
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/evaluation/integration_tests/run_infer.py b/evaluation/integration_tests/run_infer.py
@@ -233,8 +233,8 @@ def load_integration_tests() -> pd.DataFrame:
     )
     logger.info('-' * 100)
 
-    # record cost for each instance
-    df['cost'] = df['metrics'].apply(lambda x: x['accumulated_cost'])
+    # record cost for each instance, with 3 decimal places
+    df['cost'] = df['metrics'].apply(lambda x: round(x['accumulated_cost'], 3))
     logger.info(f'Total cost: USD {df["cost"].sum():.2f}')
 
     report_file = os.path.join(metadata.eval_output_dir, 'report.md')