hSterz commited on
Commit
2b1b9e2
·
1 Parent(s): b2ed338
Files changed (1) hide show
  1. src/populate.py +3 -0
src/populate.py CHANGED
@@ -24,6 +24,9 @@ def get_leaderboard_df(results_path: str, requests_path: str, cols: list, benchm
24
  if version != "1_correct":
25
  cols.remove("VCR")
26
  benchmark_cols.remove("VCR")
 
 
 
27
  df = df[cols].round(decimals=2)
28
 
29
  # filter out if any of the benchmarks have not been produced
 
24
  if version != "1_correct":
25
  cols.remove("VCR")
26
  benchmark_cols.remove("VCR")
27
+ for task in Detail_Tasks:
28
+ cols.append(task.value.col_name)
29
+ benchmark_cols.append(task.value.benchmark)
30
  df = df[cols].round(decimals=2)
31
 
32
  # filter out if any of the benchmarks have not been produced