add label candidates to sentence index edit teacher

selvamshan · Jul 27, 2018 · f6f0355 · f6f0355
1 parent 93e419c
commit f6f0355
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/parlai/tasks/squad/agents.py b/parlai/tasks/squad/agents.py
@@ -313,6 +313,7 @@ def get(self, episode_idx, entry_idx=None):
             'id': 'squad',
             'text': context + '\n' + question,
             'labels': labels,
+            'label_candidates': edited_sentences,
             'episode_done': True,
             'answer_starts': label_starts
         }