Added explicit environment closing (#129)

2025-12-17 19:20:19 +01:00 · 2018-11-22 04:25:03 -08:00
parent 16cdd9a9c1
commit 949d91321a
4 changed files with 26 additions and 0 deletions
--- a/rl_coach/environments/gym_environment.py
+++ b/rl_coach/environments/gym_environment.py
@@ -515,3 +515,11 @@ class GymEnvironment(Environment):

    def get_target_success_rate(self) -> float:
        return self.target_success_rate
+
+    def close(self) -> None:
+        """
+        Clean up to close rendering windows.
+
+        :return: None
+        """
+        self.env.close()