Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
Commit
·
ea641c7
1
Parent(s):
54e1175
fix column names
Browse files- 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=[
|
88 |
-
.sort_values(by=[AutoEvalColumn.
|
89 |
.reset_index(drop=True)
|
90 |
)
|
91 |
|
92 |
-
cols_to_round = [
|
|
|
|
|
|
|
|
|
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
|