Improve error handling and calculation in global stats and summarization functions

Author: recursix

src/agentlab/analyze/agent_xray.py

@@ -1001,9 +1001,16 @@ def get_agent_report(result_df: pd.DataFrame):
 
 
 def update_global_stats():
-    stats = inspect_results.global_report(info.result_df, reduce_fn=inspect_results.summarize_stats)
-    stats.reset_index(inplace=True)
-    return stats
+    try:
+        stats = inspect_results.global_report(
+            info.result_df, reduce_fn=inspect_results.summarize_stats
+        )
+        stats.reset_index(inplace=True)
+        return stats
+
+    except Exception as e:
+        warning(f"Error while updating global stats: {e}")
+        return None
 
 
 def update_error_report():

src/agentlab/analyze/inspect_results.py

@@ -251,7 +251,11 @@ def summarize(sub_df):
         )
     else:
         err = sub_df["err_msg"].notnull()
-        n_completed = (err | sub_df["truncated"] | sub_df["terminated"]).sum()
+        n_completed = err.copy()
+        for col in ["truncated", "terminated"]:
+            if col in sub_df:
+                n_completed = n_completed | sub_df[col]
+        n_completed = n_completed.sum()
 
         if n_completed == 0:
             return None
@@ -280,7 +284,12 @@ def summarize_stats(sub_df):
 
     # make sure there are completed runs
     err = sub_df["err_msg"].notnull()
-    n_completed = (err | sub_df["truncated"] | sub_df["terminated"]).sum()
+    n_completed = err.copy()
+    for col in ["truncated", "terminated"]:
+        if col in sub_df:
+            n_completed = n_completed | sub_df[col]
+    n_completed = n_completed.sum()
+
     if n_completed == 0:
         return None