Upload viz.py

utils/viz.py

@@ -46,7 +46,7 @@ def create_task_success_heatmap(df, benchmark_name):
     
     # Add the new row to the pivot table
     tasks_solved_df = pd.DataFrame(tasks_solved).T
-    tasks_solved_df.index = [f'<b>Tasks Solved: {tasks_solved.sum()}/{total_tasks} (All Agents)</b>']
+    tasks_solved_df.index = [f'<b>Tasks Solved: {tasks_solved.sum()}/{total_tasks} (Any Agent)</b>']
     # print number of tasks solved
     pivot_df = pd.concat([pivot_df, tasks_solved_df])