Fix squad2 setup_data (#2708)

facebookresearch · May 30, 2020 · d191b80 · d191b80
1 parent 936d1af
commit d191b80
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/parlai/tasks/squad2/agents.py b/parlai/tasks/squad2/agents.py
@@ -114,7 +114,7 @@ def setup_data(self, path):
                     ans_list = [{"text": ""}]
                     if not qa['is_impossible']:
                         ans_list = qa['answers']
-                    answers = (a['text'] for a in ans_list)
+                    answers = tuple(a['text'] for a in ans_list)
                     context = paragraph['context']
                     yield (context + '\n' + question, answers), True