hSterz commited on
Commit
5b29e88
·
1 Parent(s): ac00ce9
Files changed (1) hide show
  1. src/populate.py +3 -1
src/populate.py CHANGED
@@ -6,12 +6,14 @@ import pandas as pd
6
  from src.display.formatting import has_no_nan_values, make_clickable_model
7
  from src.display.utils import AutoEvalColumn, EvalQueueColumn
8
  from src.leaderboard.read_evals import get_raw_eval_results
 
9
 
10
 
11
  def get_leaderboard_df(results_path: str, requests_path: str, cols: list, benchmark_cols: list, version="1_correct") -> pd.DataFrame:
12
  """Creates a dataframe from all the individual experiment results"""
13
  raw_data = get_raw_eval_results(results_path+"/"+version, requests_path)
14
- all_data_json = [v.to_dict() for v in raw_data]
 
15
 
16
  print(all_data_json)
17
 
 
6
  from src.display.formatting import has_no_nan_values, make_clickable_model
7
  from src.display.utils import AutoEvalColumn, EvalQueueColumn
8
  from src.leaderboard.read_evals import get_raw_eval_results
9
+ from src.about import Tasks, N_Tasks
10
 
11
 
12
  def get_leaderboard_df(results_path: str, requests_path: str, cols: list, benchmark_cols: list, version="1_correct") -> pd.DataFrame:
13
  """Creates a dataframe from all the individual experiment results"""
14
  raw_data = get_raw_eval_results(results_path+"/"+version, requests_path)
15
+ tasks = Tasks if version == "1_correct" else N_Tasks
16
+ all_data_json = [v.to_dict(tasks) for v in raw_data]
17
 
18
  print(all_data_json)
19