Spaces:
Sleeping
Sleeping
Fix
Browse files- 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 |
-
|
|
|
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 |
|