Fix last response

THUDM · Aug 9, 2020 · f88007a · f88007a
1 parent b8408a2
commit f88007a
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/parlai/tasks/redial/agents.py b/parlai/tasks/redial/agents.py
@@ -150,6 +150,8 @@ def setup_data(self, path):
                         target_text, instance["initiatorQuestions"]
                     )
                     turn += 1
+                    if message_idx == len(messages) and target_text == "":
+                        break
                     yield (source_text, [target_text], None, [str(turn), ' '.join(previously_mentioned_movies_list + source_movie_list), ' '.join(target_movie_list), ' '.join(mentioned_entities), target_text], None), new_episode
                     new_episode = False
                     previously_mentioned_movies_list += source_movie_list + target_movie_list