Adds close method

edbeeching · Aug 3, 2023 · 680700d · 680700d
1 parent 6b73e59
commit 680700d
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/godot_rl/wrappers/clean_rl_wrapper.py b/godot_rl/wrappers/clean_rl_wrapper.py
@@ -84,6 +84,12 @@ def single_observation_space(self):
     @property
     def single_action_space(self):
         return self.envs[0].action_space
+
     @property
     def num_envs(self) -> int:
         return self.envs[0].num_envs * self.n_parallel
+
+    def close(self) -> None:
+        # Close each environment
+        for env in self.envs:
+            env.close()