hSterz commited on
Commit
0e04e71
·
1 Parent(s): b483ddc
Files changed (1) hide show
  1. src/populate.py +3 -2
src/populate.py CHANGED
@@ -26,9 +26,10 @@ def get_leaderboard_df(results_path: str, requests_path: str, cols: list, benchm
26
  df = df.sort_values(by=[AutoEvalColumn.average.name], ascending=False)
27
 
28
  print(cols)
29
- if version == "n_correct":
30
  cols.remove("VCR")
31
- benchmark_cols.remove("VCR")
 
32
  else:
33
  for task in Detail_Tasks:
34
  cols.remove(task.value.col_name)
 
26
  df = df.sort_values(by=[AutoEvalColumn.average.name], ascending=False)
27
 
28
  print(cols)
29
+ if version != "1_correct":
30
  cols.remove("VCR")
31
+ if version != "1_correct_var":
32
+ benchmark_cols.remove("VCR")
33
  else:
34
  for task in Detail_Tasks:
35
  cols.remove(task.value.col_name)