Shaltiel commited on
Commit
098b102
1 Parent(s): c2465e1

Added snli task back in

Browse files
Files changed (1) hide show
  1. 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)