Add shutdowns (#191)

parlai/core/agents.py

@@ -92,6 +92,7 @@ def shutdown(self):
         """Perform any final cleanup if needed."""
         pass
 
+
 class Teacher(Agent):
     """Basic Teacher agent which keeps track of how many times it's received
     messages. Teachers provide the ``report()`` method to get back metrics."""
@@ -150,6 +151,7 @@ def share(self):
         shared['metrics'] = self.metrics
         return shared
 
+
 class MultiTaskTeacher(Teacher):
     """Creates a teacher that is actually a set of teachers each based on
     a task string--each of these teachers will get called in turn,
@@ -259,6 +261,12 @@ def share(self):
         shared['tasks'] = [t.share() for t in self.tasks]
         return shared
 
+    def shutdown(self):
+        """Shutdown each agent."""
+        for t in self.tasks:
+            t.shutdown()
+
+
 def name_to_agent_class(name):
     words = name.split('_')
     class_name = ''

parlai/core/worlds.py

@@ -315,6 +315,7 @@ def report(self):
         return self.agents[0].report()
 
     def shutdown(self):
+        """Shutdown each agent."""
         for a in self.agents:
             a.shutdown()
 
@@ -632,6 +633,12 @@ def reset(self):
     def reset_metrics(self):
         self.worlds[0].reset_metrics()
 
+    def shutdown(self):
+        """Shutdown each world."""
+        for w in self.worlds:
+            w.shutdown()
+        self.world.shutdown()
+
 
 class HogwildProcess(Process):
     """Process child used for ``HogwildWorld``.