nan commited on
Commit
6c22ed3
1 Parent(s): 097c6d1

chore: clean up

Browse files
src/display/utils.py CHANGED
@@ -51,21 +51,12 @@ AutoEvalColumnLongDoc = make_autoevalcolumn(
51
  "AutoEvalColumnLongDoc", BenchmarksLongDoc)
52
 
53
 
54
- ## For the queue columns in the submission tab
55
- @dataclass(frozen=True)
56
- class EvalQueueColumn: # Queue column
57
- model = ColumnContent("model", "markdown", True)
58
- status = ColumnContent("status", "str", True)
59
-
60
-
61
  # Column selection
62
  COLS_QA = [c.name for c in fields(AutoEvalColumnQA) if not c.hidden]
63
  COLS_LONG_DOC = [c.name for c in fields(AutoEvalColumnLongDoc) if not c.hidden]
64
  TYPES = [c.type for c in fields(AutoEvalColumnQA) if not c.hidden]
65
  COLS_LITE = [c.name for c in fields(AutoEvalColumnQA) if c.displayed_by_default and not c.hidden]
66
 
67
- EVAL_COLS = [c.name for c in fields(EvalQueueColumn)]
68
-
69
  QA_BENCHMARK_COLS = [t.value.col_name for t in BenchmarksQA]
70
 
71
  LONG_DOC_BENCHMARK_COLS = [t.value.col_name for t in BenchmarksLongDoc]
 
51
  "AutoEvalColumnLongDoc", BenchmarksLongDoc)
52
 
53
 
 
 
 
 
 
 
 
54
  # Column selection
55
  COLS_QA = [c.name for c in fields(AutoEvalColumnQA) if not c.hidden]
56
  COLS_LONG_DOC = [c.name for c in fields(AutoEvalColumnLongDoc) if not c.hidden]
57
  TYPES = [c.type for c in fields(AutoEvalColumnQA) if not c.hidden]
58
  COLS_LITE = [c.name for c in fields(AutoEvalColumnQA) if c.displayed_by_default and not c.hidden]
59
 
 
 
60
  QA_BENCHMARK_COLS = [t.value.col_name for t in BenchmarksQA]
61
 
62
  LONG_DOC_BENCHMARK_COLS = [t.value.col_name for t in BenchmarksLongDoc]
tests/src/display/test_utils.py CHANGED
@@ -1,5 +1,5 @@
1
  import pytest
2
- from src.display.utils import fields, AutoEvalColumnQA, AutoEvalColumnLongDoc, COLS_QA, COLS_LONG_DOC, COLS_LITE, TYPES, EVAL_COLS, QA_BENCHMARK_COLS, LONG_DOC_BENCHMARK_COLS
3
 
4
 
5
  def test_fields():
@@ -12,6 +12,5 @@ def test_macro_variables():
12
  print(f'COLS_LONG_DOC: {COLS_LONG_DOC}')
13
  print(f'COLS_LITE: {COLS_LITE}')
14
  print(f'TYPES: {TYPES}')
15
- print(f'EVAL_COLS: {EVAL_COLS}')
16
  print(f'QA_BENCHMARK_COLS: {QA_BENCHMARK_COLS}')
17
  print(f'LONG_DOC_BENCHMARK_COLS: {LONG_DOC_BENCHMARK_COLS}')
 
1
  import pytest
2
+ from src.display.utils import fields, AutoEvalColumnQA, COLS_QA, COLS_LONG_DOC, COLS_LITE, TYPES, QA_BENCHMARK_COLS, LONG_DOC_BENCHMARK_COLS
3
 
4
 
5
  def test_fields():
 
12
  print(f'COLS_LONG_DOC: {COLS_LONG_DOC}')
13
  print(f'COLS_LITE: {COLS_LITE}')
14
  print(f'TYPES: {TYPES}')
 
15
  print(f'QA_BENCHMARK_COLS: {QA_BENCHMARK_COLS}')
16
  print(f'LONG_DOC_BENCHMARK_COLS: {LONG_DOC_BENCHMARK_COLS}')