From 64833c6c72b61a3295eb3b7104f2bda65a5477cb Mon Sep 17 00:00:00 2001 From: Kiran Karra Date: Thu, 20 Aug 2020 17:48:57 -0400 Subject: [PATCH] adds explicit killing of workers --- torch_ac/utils/penv.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/torch_ac/utils/penv.py b/torch_ac/utils/penv.py index 6e4f4f1..6e5c9f1 100644 --- 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))