karimouda commited on
Commit
f79393b
·
1 Parent(s): 8feff4f
Files changed (1) hide show
  1. src/populate.py +9 -6
src/populate.py CHANGED
@@ -15,12 +15,15 @@ def get_leaderboard_df(results_path: str, requests_path: str, cols: list, benchm
15
 
16
  df = pd.DataFrame.from_records(all_data_json)
17
  print(df)
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
 
 
 
24
 
25
 
26
  def get_evaluation_queue_df(save_path: str, cols: list) -> list[pd.DataFrame]:
 
15
 
16
  df = pd.DataFrame.from_records(all_data_json)
17
  print(df)
18
+ if not df.empty:
19
+ df = df.sort_values(by=[AutoEvalColumn.average.name], ascending=False)
20
+ df = df[cols].round(decimals=2)
21
+
22
+ # filter out if any of the benchmarks have not been produced
23
+ df = df[has_no_nan_values(df, benchmark_cols)]
24
+ return df
25
+ else:
26
+ return pd.DataFrame(columns=cols)
27
 
28
 
29
  def get_evaluation_queue_df(save_path: str, cols: list) -> list[pd.DataFrame]: