Spaces:
Paused
Paused
Added snli task back in
Browse files- custom_tasks.py +2 -1
custom_tasks.py
CHANGED
@@ -10,11 +10,12 @@ from src.custom_tasks.heq_task import *
|
|
10 |
from src.custom_tasks.sentiment_task import *
|
11 |
from src.custom_tasks.winograd_task import *
|
12 |
from src.custom_tasks.translation_task import *
|
|
|
13 |
|
14 |
## MODULE LOGIC
|
15 |
# You should not need to touch this
|
16 |
# Convert to dict for lighteval
|
17 |
-
TASKS_TABLE = [task.as_dict() for task in [heq_task, sentiment_task, winograd_task, translation_task]]
|
18 |
|
19 |
if __name__ == "__main__":
|
20 |
print(t["name"] for t in TASKS_TABLE)
|
|
|
10 |
from src.custom_tasks.sentiment_task import *
|
11 |
from src.custom_tasks.winograd_task import *
|
12 |
from src.custom_tasks.translation_task import *
|
13 |
+
from src.custom_tasks.snli_task import *
|
14 |
|
15 |
## MODULE LOGIC
|
16 |
# You should not need to touch this
|
17 |
# Convert to dict for lighteval
|
18 |
+
TASKS_TABLE = [task.as_dict() for task in [heq_task, sentiment_task, winograd_task, translation_task, snli_task]]
|
19 |
|
20 |
if __name__ == "__main__":
|
21 |
print(t["name"] for t in TASKS_TABLE)
|