Alvinn-aai commited on
Commit
ea641c7
·
1 Parent(s): 54e1175

fix column names

Browse files
Files changed (1) hide show
  1. src/populate.py +7 -3
src/populate.py CHANGED
@@ -84,12 +84,16 @@ def get_leaderboard_df(results_dataset_name: str) -> pd.DataFrame:
84
  # Keep only the latest entry per unique (System Name, System Type, Organization) triplet
85
  final_df = (
86
  full_df.sort_values("Submitted On", ascending=False)
87
- .drop_duplicates(subset=["System Name", "System Type", "Organization"], keep="first")
88
- .sort_values(by=[AutoEvalColumn.success_rate.name], ascending=False)
89
  .reset_index(drop=True)
90
  )
91
 
92
- cols_to_round = ["Success Rate (%)"]
 
 
 
 
93
  final_df[cols_to_round] = final_df[cols_to_round].round(decimals=2)
94
 
95
  return final_df
 
84
  # Keep only the latest entry per unique (System Name, System Type, Organization) triplet
85
  final_df = (
86
  full_df.sort_values("Submitted On", ascending=False)
87
+ .drop_duplicates(subset=[AutoEvalColumn.system.name, AutoEvalColumn.organization.name], keep="first")
88
+ .sort_values(by=[AutoEvalColumn.success_rate_overall.name], ascending=False)
89
  .reset_index(drop=True)
90
  )
91
 
92
+ cols_to_round = [
93
+ AutoEvalColumn.success_rate_overall.name,
94
+ AutoEvalColumn.success_rate_tier1.name,
95
+ AutoEvalColumn.success_rate_tier2.name,
96
+ ]
97
  final_df[cols_to_round] = final_df[cols_to_round].round(decimals=2)
98
 
99
  return final_df