add env.close method

this helps clean up the render windows as mentioned here openai/gym#1056 (comment)
bionicles · Jun 29, 2020 · d45fc93 · d45fc93
1 parent 453787a
commit d45fc93
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/slimevolleygym/slimevolley.py b/slimevolleygym/slimevolley.py
@@ -867,6 +867,10 @@ def render(self, mode='human', close=False):
       self.game.display(self.viewer)
       return self.viewer.render(return_rgb_array = mode=='rgb_array')
 
+  def close(self):
+    if self.viewer():
+      self.viewer.close()
+
   def get_action_meanings(self):
     return [self.atari_action_meaning[i] for i in self.atari_action_set]