Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
chore: clean up
Browse files- src/display/utils.py +0 -9
- tests/src/display/test_utils.py +1 -2
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,
|
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}')
|