jasonshaoshun commited on
Commit
1014997
·
1 Parent(s): 0101cf1
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -382,7 +382,6 @@ def init_leaderboard_mib_subgraph(dataframe, track):
382
  field_name = f"{task.value.benchmark}_{model}"
383
  display_name = f"{benchmark_mapping[task.value.benchmark]}({model_name_mapping[model]})"
384
  display_mapping[field_name] = display_name
385
- print("\nDisplay name mapping:", display_mapping)
386
 
387
 
388
  # Now when creating benchmark groups, we'll use display names
@@ -420,7 +419,7 @@ def init_leaderboard_mib_subgraph(dataframe, track):
420
  # Important: We need to rename our DataFrame columns to match display names
421
 
422
  renamed_df = dataframe.rename(columns=display_mapping)
423
- all_columns = dataframe.columns.tolist()
424
 
425
  # Original code
426
  return Leaderboard(
 
382
  field_name = f"{task.value.benchmark}_{model}"
383
  display_name = f"{benchmark_mapping[task.value.benchmark]}({model_name_mapping[model]})"
384
  display_mapping[field_name] = display_name
 
385
 
386
 
387
  # Now when creating benchmark groups, we'll use display names
 
419
  # Important: We need to rename our DataFrame columns to match display names
420
 
421
  renamed_df = dataframe.rename(columns=display_mapping)
422
+ all_columns = [c.name for c in fields(AutoEvalColumn_mib_subgraph) if c.displayed_by_default],
423
 
424
  # Original code
425
  return Leaderboard(