lixuejing commited on
Commit
19e21f2
Β·
1 Parent(s): c3ddec5
Files changed (2) hide show
  1. app.py +1 -0
  2. src/display/utils.py +1 -0
app.py CHANGED
@@ -18,6 +18,7 @@ from src.display.utils import (
18
  COLS,
19
  EVAL_COLS,
20
  EVAL_TYPES,
 
21
  AutoEvalColumn,
22
  ModelType,
23
  fields,
 
18
  COLS,
19
  EVAL_COLS,
20
  EVAL_TYPES,
21
+ TYPES,
22
  AutoEvalColumn,
23
  ModelType,
24
  fields,
src/display/utils.py CHANGED
@@ -119,6 +119,7 @@ class Precision(Enum):
119
 
120
  # Column selection
121
  COLS = [c.name for c in fields(AutoEvalColumn) if not c.hidden]
 
122
 
123
  EVAL_COLS = [c.name for c in fields(EvalQueueColumn)]
124
  EVAL_TYPES = [c.type for c in fields(EvalQueueColumn)]
 
119
 
120
  # Column selection
121
  COLS = [c.name for c in fields(AutoEvalColumn) if not c.hidden]
122
+ TYPES = [c.type for c in fields(AutoEvalColumn) if not c.hidden]
123
 
124
  EVAL_COLS = [c.name for c in fields(EvalQueueColumn)]
125
  EVAL_TYPES = [c.type for c in fields(EvalQueueColumn)]