Merge pull request facebookresearch#700 from facebookresearch/msgr_mn…

…gr_fix Messenger manager bug fix
analysmith · Apr 16, 2018 · c5f6c45 · c5f6c45
2 parents ad98e94 + 1d1a082
commit c5f6c45
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/parlai/messenger/core/messenger_manager.py b/parlai/messenger/core/messenger_manager.py
@@ -568,7 +568,8 @@ def _task_function(opt, agents, conversation_id, world_type):
                         agents = [a for a in agents if a.disp_id is not None]
                         for a in agents:
                             # Remove selected workers from the agent pool
-                            self.remove_agent_from_pool(self._get_agent_state(a.id), world_type=world_type)
+                            self.remove_agent_from_pool(self._get_agent_state(a.id), \
+                                    world_type=world_type, mark_removed=False)
                         for a in agents:
                             partner_list = agents.copy()
                             partner_list.remove(a)