Behnamm commited on
Commit
4210493
1 Parent(s): 1f4d57a

Update src/populate.py

Browse files
Files changed (1) hide show
  1. src/populate.py +2 -1
src/populate.py CHANGED
@@ -15,7 +15,8 @@ def get_leaderboard_df(results_path: str, requests_path: str, cols: list, benchm
15
 
16
  df = pd.DataFrame.from_records(all_data_json)
17
  df = df.sort_values(by=[AutoEvalColumn.average.name], ascending=False) if AutoEvalColumn.average.name in df.columns else df
18
- df = df[cols].round(decimals=2)
 
19
 
20
  # filter out if any of the benchmarks have not been produced
21
  df = df[has_no_nan_values(df, benchmark_cols)]
 
15
 
16
  df = pd.DataFrame.from_records(all_data_json)
17
  df = df.sort_values(by=[AutoEvalColumn.average.name], ascending=False) if AutoEvalColumn.average.name in df.columns else df
18
+ filtered_cols = [col for col in cols if col in df.columns]
19
+ df = df[filtered_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)]