fix test

recursix · gasse · commit ef33f1f76caf · 2024-11-11T09:46:34.000-05:00
diff --git a/src/agentlab/analyze/inspect_results.py b/src/agentlab/analyze/inspect_results.py
@@ -295,8 +295,9 @@ def summarize(sub_df, use_bootstrap=False):
             avg_steps=sub_df["n_steps"].mean(skipna=True).round(3),
             n_completed=f"{n_completed}/{len(sub_df)}",
             n_err=err.sum(skipna=True),
-            cum_cost=sub_df["stats.cum_cost"].sum(skipna=True).round(4),
         )
+        if "stats.cum_cost" in sub_df:
+            record["cum_cost"]=sub_df["stats.cum_cost"].sum(skipna=True).round(4),
 
     return pd.Series(record)
 

Original file line number	Diff line number	Diff line change
`@@ -295,8 +295,9 @@ def summarize(sub_df, use_bootstrap=False):`
`295`	`295`	`avg_steps=sub_df["n_steps"].mean(skipna=True).round(3),`
`296`	`296`	`n_completed=f"{n_completed}/{len(sub_df)}",`
`297`	`297`	`n_err=err.sum(skipna=True),`
`298`		`- cum_cost=sub_df["stats.cum_cost"].sum(skipna=True).round(4),`
`299`	`298`	`)`
	`299`	`+ if "stats.cum_cost" in sub_df:`
	`300`	`+ record["cum_cost"]=sub_df["stats.cum_cost"].sum(skipna=True).round(4),`
`300`	`301`
`301`	`302`	`return pd.Series(record)`
`302`	`303`