sh1gechan commited on
Commit
a02d2a5
·
verified ·
1 Parent(s): 95eda9e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +13 -5
app.py CHANGED
@@ -121,7 +121,8 @@ def select_columns(df: pd.DataFrame, columns: list) -> pd.DataFrame:
121
  AutoEvalColumn.model_type_symbol.name,
122
  AutoEvalColumn.model.name,
123
  ]
124
- selected_cols = always_here_cols + [c for c in columns if c in df.columns]
 
125
  return df[selected_cols]
126
 
127
 
@@ -269,14 +270,21 @@ with demo:
269
  )
270
 
271
  leaderboard_df_filtered = filter_models(leaderboard_df, [t.to_str(" : ") for t in ModelType], list(NUMERIC_INTERVALS.keys()), [i.value.name for i in Precision], [i.value.name for i in AddSpecialTokens], [i.value.name for i in NumFewShots], False, False, False)
272
- initial_columns = [c.name for c in fields(AutoEvalColumn) if c.never_hidden] + shown_columns.value
273
  leaderboard_df_filtered = select_columns(leaderboard_df_filtered, initial_columns)
274
-
275
 
 
 
 
 
 
 
 
 
276
  leaderboard_table = gr.components.Dataframe(
277
  value=leaderboard_df_filtered,
278
- headers=[c.name for c in fields(AutoEvalColumn) if c.never_hidden] + shown_columns.value,
279
- datatype=TYPES,
280
  elem_id="leaderboard-table",
281
  interactive=False,
282
  visible=True,
 
121
  AutoEvalColumn.model_type_symbol.name,
122
  AutoEvalColumn.model.name,
123
  ]
124
+
125
+ selected_cols = list(dict.fromkeys(always_here_cols + [c for c in columns if c in df.columns]))
126
  return df[selected_cols]
127
 
128
 
 
270
  )
271
 
272
  leaderboard_df_filtered = filter_models(leaderboard_df, [t.to_str(" : ") for t in ModelType], list(NUMERIC_INTERVALS.keys()), [i.value.name for i in Precision], [i.value.name for i in AddSpecialTokens], [i.value.name for i in NumFewShots], False, False, False)
273
+ initial_columns = [c.name for c in fields(AutoEvalColumn) if c.never_hidden or c.displayed_by_default]
274
  leaderboard_df_filtered = select_columns(leaderboard_df_filtered, initial_columns)
 
275
 
276
+ # leaderboard_table = gr.components.Dataframe(
277
+ # value=leaderboard_df_filtered,
278
+ # headers=[c.name for c in fields(AutoEvalColumn) if c.never_hidden] + shown_columns.value,
279
+ # datatype=TYPES,
280
+ # elem_id="leaderboard-table",
281
+ # interactive=False,
282
+ # visible=True,
283
+ # )
284
  leaderboard_table = gr.components.Dataframe(
285
  value=leaderboard_df_filtered,
286
+ headers=list(leaderboard_df_filtered.columns),
287
+ datatype={col: TYPES[COLS.index(col)] if col in COLS else "str" for col in leaderboard_df_filtered.columns},
288
  elem_id="leaderboard-table",
289
  interactive=False,
290
  visible=True,