jasonshaoshun commited on
Commit
4a47622
·
1 Parent(s): f6dfabd
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -392,11 +392,11 @@ def init_leaderboard_mib_subgraph(dataframe, track):
392
  raise ValueError("Leaderboard DataFrame is empty or None.")
393
 
394
  # Print DataFrame information for debugging
395
- print("\nDebugging DataFrame columns:", dataframe.columns.tolist())
396
 
397
  # Get result columns (excluding Method and Average)
398
  result_columns = [col for col in dataframe.columns
399
- if col not in ['Method', 'Average'] and '_' in col]
400
 
401
  # Create benchmark and model selections
402
  benchmarks = set()
@@ -406,7 +406,7 @@ def init_leaderboard_mib_subgraph(dataframe, track):
406
  # Extract unique benchmarks and models from column names
407
  for col in result_columns:
408
  print(f"col is {col}")
409
- benchmark, model = col.split('_')
410
  benchmarks.add(benchmark)
411
  models.add(model)
412
  print(f"benchmark is {benchmark} and model is {model}")
 
392
  raise ValueError("Leaderboard DataFrame is empty or None.")
393
 
394
  # Print DataFrame information for debugging
395
+ # print("\nDebugging DataFrame columns:", dataframe.columns.tolist())
396
 
397
  # Get result columns (excluding Method and Average)
398
  result_columns = [col for col in dataframe.columns
399
+ if col not in ['Method', 'Average', 'eval_name'] and '_' in col]
400
 
401
  # Create benchmark and model selections
402
  benchmarks = set()
 
406
  # Extract unique benchmarks and models from column names
407
  for col in result_columns:
408
  print(f"col is {col}")
409
+ benchmark, model = col.split('_', maxsplit=1)
410
  benchmarks.add(benchmark)
411
  models.add(model)
412
  print(f"benchmark is {benchmark} and model is {model}")