Aaron Mueller commited on
Commit
0273423
·
1 Parent(s): 5833731

filtering by track

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -61,7 +61,7 @@ def init_leaderboard(dataframe, track):
61
  if dataframe is None or dataframe.empty:
62
  raise ValueError("Leaderboard DataFrame is empty or None.")
63
  # filter for correct track
64
- dataframe = dataframe.loc[dataframe["track"] == track]
65
  return Leaderboard(
66
  value=dataframe,
67
  datatype=[c.type for c in fields(AutoEvalColumn)],
 
61
  if dataframe is None or dataframe.empty:
62
  raise ValueError("Leaderboard DataFrame is empty or None.")
63
  # filter for correct track
64
+ dataframe = dataframe.loc[dataframe["eval_name"].endswith(track.lower())]
65
  return Leaderboard(
66
  value=dataframe,
67
  datatype=[c.type for c in fields(AutoEvalColumn)],