sh1gechan commited on
Commit
aef403c
·
verified ·
1 Parent(s): b6e84a7

Update src/populate.py

Browse files
Files changed (1) hide show
  1. src/populate.py +1 -9
src/populate.py CHANGED
@@ -13,19 +13,11 @@ def get_leaderboard_df(results_path: str, requests_path: str, cols: list, benchm
13
  raw_data = get_raw_eval_results(results_path, requests_path)
14
  all_data_json = [v.to_dict() for v in raw_data]
15
 
16
- print("Keys in the first dictionary:")
17
- print(list(all_data_json[0].keys()))
18
 
19
  df = pd.DataFrame.from_records(all_data_json)
20
- # df = df.sort_values(by=[AutoEvalColumn.average.name], ascending=False)
21
  df = df[cols].round(decimals=2)
22
 
23
- print("Columns in the DataFrame:")
24
- print(df.columns.tolist())
25
-
26
- print("Requested columns:")
27
- print(cols)
28
-
29
  # filter out if any of the benchmarks have not been produced
30
  df = df[has_no_nan_values(df, benchmark_cols)]
31
  return df
 
13
  raw_data = get_raw_eval_results(results_path, requests_path)
14
  all_data_json = [v.to_dict() for v in raw_data]
15
 
 
 
16
 
17
  df = pd.DataFrame.from_records(all_data_json)
18
+ df = df.sort_values(by=[AutoEvalColumn.average.name], ascending=False)
19
  df = df[cols].round(decimals=2)
20
 
 
 
 
 
 
 
21
  # filter out if any of the benchmarks have not been produced
22
  df = df[has_no_nan_values(df, benchmark_cols)]
23
  return df