benediktstroebl commited on
Commit
4bb1605
·
1 Parent(s): 512799d

removed print

Browse files
Files changed (1) hide show
  1. utils/viz.py +1 -1
utils/viz.py CHANGED
@@ -41,7 +41,7 @@ def create_task_success_heatmap(df, benchmark_name):
41
  tasks_solved = (pivot_df.sum(axis=0) > 0).astype(int)
42
  # Total number of tasks (columns)
43
  total_tasks = len(pivot_df.columns)
44
- print(benchmark_name)
45
  if benchmark_name == "SWE-bench Verified (Mini)":
46
  total_tasks = 50 # TODO - remove hardcoding
47
 
 
41
  tasks_solved = (pivot_df.sum(axis=0) > 0).astype(int)
42
  # Total number of tasks (columns)
43
  total_tasks = len(pivot_df.columns)
44
+
45
  if benchmark_name == "SWE-bench Verified (Mini)":
46
  total_tasks = 50 # TODO - remove hardcoding
47