Spaces:
Running
Running
benediktstroebl
commited on
Commit
·
4bb1605
1
Parent(s):
512799d
removed print
Browse files- 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 |
-
|
| 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 |
|