Undo change to test

tests/test_evaluator.py

@@ -8,7 +8,7 @@ import pytest
 # TODO: more fine grained unit tests rather than this big honking integration
 # test once we break evaluator into smaller, more manageable pieces
 
-@pytest.mark.parametrize("taskname,Task", [('squad', tasks.squad.SQuAD)])
+@pytest.mark.parametrize("taskname,Task", tasks.TASK_REGISTRY.items())
 def test_evaluator(taskname, Task):
     task_dict = tasks.get_task_dict([taskname])
     lm = models.get_model('dummy')()
@@ -29,4 +29,4 @@ def test_evaluator(taskname, Task):
         
 
     lm.loglikelihood = ll_fn
-    evaluator.evaluate(lm, task_dict, False, 0, 10)
\ No newline at end of file
+    evaluator.evaluate(lm, task_dict, False, 0, 10)