abhinav-joshi commited on
Commit
bcd68d3
·
1 Parent(s): 58c47f7

add model name and symbol to leaderboard

Browse files
Files changed (2) hide show
  1. app.py +6 -6
  2. 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
- # 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
 
 
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
- # 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),
 
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),