jasonshaoshun commited on
Commit
009db1f
·
1 Parent(s): f732437
Files changed (2) hide show
  1. app.py +2 -2
  2. src/display/utils.py +1 -1
app.py CHANGED
@@ -413,13 +413,13 @@ def init_leaderboard_mib_subgraph(dataframe, track):
413
  # Create selection groups
414
  benchmark_selections = {
415
  # For each benchmark, store which columns should be shown
416
- benchmark: [col for col in result_columns if col.startswith(f"{benchmark}(")]
417
  for benchmark in benchmarks
418
  }
419
 
420
  model_selections = {
421
  # For each model, store which columns should be shown
422
- model: [col for col in result_columns if col.endswith(f"({model})")]
423
  for model in models
424
  }
425
 
 
413
  # Create selection groups
414
  benchmark_selections = {
415
  # For each benchmark, store which columns should be shown
416
+ benchmark: [col for col in result_columns if col.startswith(f"{benchmark}_")]
417
  for benchmark in benchmarks
418
  }
419
 
420
  model_selections = {
421
  # For each model, store which columns should be shown
422
+ model: [col for col in result_columns if col.startswith(f"_{model}")]
423
  for model in models
424
  }
425
 
src/display/utils.py CHANGED
@@ -132,7 +132,7 @@ auto_eval_column_dict_mib_subgraph.append(
132
  for task in TasksMib_Subgraph:
133
  for model in task.value.models:
134
  # Create a valid field name by using underscores
135
- field_name = f"{task.value.benchmark}-{model}"
136
  # The display name can be more descriptive
137
  display_name = f"{task.value.benchmark}({model})"
138
 
 
132
  for task in TasksMib_Subgraph:
133
  for model in task.value.models:
134
  # Create a valid field name by using underscores
135
+ field_name = f"{task.value.benchmark}_{model}"
136
  # The display name can be more descriptive
137
  display_name = f"{task.value.benchmark}({model})"
138