update about.py
Browse files- src/about.py +14 -1
src/about.py
CHANGED
@@ -9,7 +9,19 @@ class TaskInfo:
|
|
9 |
metric: str
|
10 |
|
11 |
|
12 |
-
#
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
13 |
class Tasks(Enum):
|
14 |
Professional_Law = TaskInfo(benchmark='professional_law', col_name='Professional Law', metric='accuracy')
|
15 |
Moral_Scenarios = TaskInfo(benchmark='moral_scenarios', col_name='Moral Scenarios', metric='accuracy')
|
@@ -62,6 +74,7 @@ class Tasks(Enum):
|
|
62 |
Medical_Genetics = TaskInfo(benchmark='medical_genetics', col_name='Medical Genetics', metric='accuracy')
|
63 |
Abstract_Algebra = TaskInfo(benchmark='abstract_algebra', col_name='Abstract Algebra', metric='accuracy')
|
64 |
|
|
|
65 |
# Now include the variables expected by app.py
|
66 |
|
67 |
TITLE = """
|
|
|
9 |
metric: str
|
10 |
|
11 |
|
12 |
+
# src/about.py
|
13 |
+
|
14 |
+
from enum import Enum
|
15 |
+
from dataclasses import dataclass
|
16 |
+
|
17 |
+
# Define TaskInfo dataclass
|
18 |
+
@dataclass
|
19 |
+
class TaskInfo:
|
20 |
+
benchmark: str
|
21 |
+
col_name: str
|
22 |
+
metric: str
|
23 |
+
|
24 |
+
# Define Tasks enum with your specific subjects, excluding the unwanted ones
|
25 |
class Tasks(Enum):
|
26 |
Professional_Law = TaskInfo(benchmark='professional_law', col_name='Professional Law', metric='accuracy')
|
27 |
Moral_Scenarios = TaskInfo(benchmark='moral_scenarios', col_name='Moral Scenarios', metric='accuracy')
|
|
|
74 |
Medical_Genetics = TaskInfo(benchmark='medical_genetics', col_name='Medical Genetics', metric='accuracy')
|
75 |
Abstract_Algebra = TaskInfo(benchmark='abstract_algebra', col_name='Abstract Algebra', metric='accuracy')
|
76 |
|
77 |
+
|
78 |
# Now include the variables expected by app.py
|
79 |
|
80 |
TITLE = """
|