hSterz commited on
Commit
a01b95f
·
1 Parent(s): ba0dc5b
Files changed (1) hide show
  1. src/populate.py +3 -2
src/populate.py CHANGED
@@ -24,9 +24,10 @@ 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
  for task in Detail_Tasks:
28
- cols.append(task.value.col_name)
29
- benchmark_cols.append(task.value.benchmark)
30
  print(df)
31
  print(cols)
32
  df = df[cols].round(decimals=2)
 
24
  if version != "1_correct":
25
  cols.remove("VCR")
26
  benchmark_cols.remove("VCR")
27
+ else:
28
  for task in Detail_Tasks:
29
+ cols.remove(task.value.col_name)
30
+ benchmark_cols.remove(task.value.benchmark)
31
  print(df)
32
  print(cols)
33
  df = df[cols].round(decimals=2)