Fix a bug for single process

2026-09-09 11:13:47 +08:00 · 2018-05-25 19:52:58 -06:00
parent d7c537707a
commit 5e91086c17
1 changed files with 3 additions and 0 deletions
@@ -196,6 +196,7 @@ class ParallelizedTask:
        self.state_dim = self.tasks[0].state_dim
        self.action_dim = self.tasks[0].action_dim
        self.name = self.tasks[0].name
+        self.single_process = single_process

    def step(self, actions):
        results = [task.step(action) for task, action in zip(self.tasks, actions)]
@@ -207,4 +208,6 @@ class ParallelizedTask:
        return np.stack(results)

    def close(self):
+        if self.single_process:
+            return
        for task in self.tasks: task.close()