HungryTorch commited on
Commit
905c5d0
·
verified ·
1 Parent(s): 63f7a04

Show Elo on home cards

Browse files
Files changed (2) hide show
  1. leaderboard.py +7 -3
  2. render.py +1 -1
leaderboard.py CHANGED
@@ -44,10 +44,10 @@ ELO_REGULARIZATION = 0.1
44
 
45
  @dataclass(frozen=True)
46
  class TopModel:
47
- """One line of a home card's mini-ranking; ``score`` is None when unrankable."""
48
 
49
  name: str
50
- score: float | None
51
  is_baseline: bool
52
 
53
 
@@ -337,7 +337,11 @@ def top_models(
337
  return [
338
  TopModel(
339
  name=str(e["model_name"]),
340
- score=_round(e["rank"]),
 
 
 
 
341
  is_baseline=e["is_baseline"],
342
  )
343
  for e in _entries(df, by_id, board)[:limit]
 
44
 
45
  @dataclass(frozen=True)
46
  class TopModel:
47
+ """One line of a home card's mini-ranking; ``elo`` is None when unavailable."""
48
 
49
  name: str
50
+ elo: int | None
51
  is_baseline: bool
52
 
53
 
 
337
  return [
338
  TopModel(
339
  name=str(e["model_name"]),
340
+ elo=(
341
+ round(e["elo"])
342
+ if math.isfinite(e["elo"]) and math.isfinite(e["rank"])
343
+ else None
344
+ ),
345
  is_baseline=e["is_baseline"],
346
  )
347
  for e in _entries(df, by_id, board)[:limit]
render.py CHANGED
@@ -123,7 +123,7 @@ def _leading(top) -> str:
123
  if row.is_baseline
124
  else ""
125
  )
126
- score = "n/a" if row.score is None else f"{row.score:.3f}"
127
  parts.append(
128
  f'<p class="pm-leader">{escape(row.name)}{badge}'
129
  f' <span class="pm-num--dim">{score}</span></p>'
 
123
  if row.is_baseline
124
  else ""
125
  )
126
+ score = "n/a" if row.elo is None else str(row.elo)
127
  parts.append(
128
  f'<p class="pm-leader">{escape(row.name)}{badge}'
129
  f' <span class="pm-num--dim">{score}</span></p>'