Adam Jirkovsky commited on
Commit
cd24b99
·
1 Parent(s): b2e7d0b

Fix BENCHMARK_COLS

Browse files
Files changed (1) hide show
  1. src/display/utils.py +1 -1
src/display/utils.py CHANGED
@@ -179,7 +179,7 @@ TYPES_LITE = [c.type for c in fields(AutoEvalColumn) if c.displayed_by_default a
179
  EVAL_COLS = [c.name for c in fields(EvalQueueColumn)]
180
  EVAL_TYPES = [c.type for c in fields(EvalQueueColumn)]
181
 
182
- BENCHMARK_COLS = [t.value.col_name for t in Tasks]
183
 
184
  NUMERIC_INTERVALS = {
185
  "?": pd.Interval(-1, 0, closed="right"),
 
179
  EVAL_COLS = [c.name for c in fields(EvalQueueColumn)]
180
  EVAL_TYPES = [c.type for c in fields(EvalQueueColumn)]
181
 
182
+ BENCHMARK_COLS = [HEADER_MAP[t.value.col_name] for t in Tasks]
183
 
184
  NUMERIC_INTERVALS = {
185
  "?": pd.Interval(-1, 0, closed="right"),