Commit
·
bcd68d3
1
Parent(s):
58c47f7
add model name and symbol to leaderboard
Browse files- app.py +6 -6
- src/leaderboard/read_evals.py +2 -2
app.py
CHANGED
@@ -92,13 +92,13 @@ def search_table(df: pd.DataFrame, query: str) -> pd.DataFrame:
|
|
92 |
|
93 |
|
94 |
def select_columns(df: pd.DataFrame, columns: list) -> pd.DataFrame:
|
95 |
-
|
96 |
-
|
97 |
-
|
98 |
-
|
99 |
# We use COLS to maintain sorting
|
100 |
-
|
101 |
-
filtered_df = df[[c for c in COLS if c in df.columns and c in columns]]
|
102 |
return filtered_df
|
103 |
|
104 |
|
|
|
92 |
|
93 |
|
94 |
def select_columns(df: pd.DataFrame, columns: list) -> pd.DataFrame:
|
95 |
+
always_here_cols = [
|
96 |
+
AutoEvalColumn.model_type_symbol.name,
|
97 |
+
AutoEvalColumn.model.name,
|
98 |
+
]
|
99 |
# We use COLS to maintain sorting
|
100 |
+
filtered_df = df[always_here_cols + [c for c in COLS if c in df.columns and c in columns]]
|
101 |
+
# filtered_df = df[[c for c in COLS if c in df.columns and c in columns]]
|
102 |
return filtered_df
|
103 |
|
104 |
|
src/leaderboard/read_evals.py
CHANGED
@@ -115,8 +115,8 @@ class EvalResult:
|
|
115 |
data_dict = {
|
116 |
"eval_name": self.eval_name, # not a column, just a save name,
|
117 |
# AutoEvalColumn.precision.name: self.precision.value.name,
|
118 |
-
|
119 |
-
|
120 |
# AutoEvalColumn.weight_type.name: self.weight_type.value.name,
|
121 |
# AutoEvalColumn.architecture.name: self.architecture,
|
122 |
# AutoEvalColumn.model.name: make_clickable_model(self.full_model),
|
|
|
115 |
data_dict = {
|
116 |
"eval_name": self.eval_name, # not a column, just a save name,
|
117 |
# AutoEvalColumn.precision.name: self.precision.value.name,
|
118 |
+
AutoEvalColumn.model_type.name: self.model_type.value.name,
|
119 |
+
AutoEvalColumn.model_type_symbol.name: self.model_type.value.symbol,
|
120 |
# AutoEvalColumn.weight_type.name: self.weight_type.value.name,
|
121 |
# AutoEvalColumn.architecture.name: self.architecture,
|
122 |
# AutoEvalColumn.model.name: make_clickable_model(self.full_model),
|