adds explicit killing of workers

lcswillems · Aug 20, 2020 · 64833c6 · 64833c6
1 parent e7fd252
commit 64833c6
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/torch_ac/utils/penv.py b/torch_ac/utils/penv.py
@@ -12,6 +12,8 @@ def worker(conn, env):
         elif cmd == "reset":
             obs = env.reset()
             conn.send(obs)
+        elif cmd == "kill":
+            return
         else:
             raise NotImplementedError
 
@@ -34,6 +36,10 @@ def __init__(self, envs):
             p.start()
             remote.close()
 
+    def __del__(self):
+        for local in self.locals:
+            local.send(("kill", None))
+
     def reset(self):
         for local in self.locals:
             local.send(("reset", None))