Merge pull request #4 from kkarrancsu/master

Explicitly kill workers when ParallelEnv instance is destroyed
lcswillems · Aug 21, 2020 · d4212e1 · d4212e1
2 parents e7fd252 + 64833c6
commit d4212e1
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))