Browse Source

option to set rendering_mode when creating the environment with gym.make, as required by the new gym API

saleml 2 năm trước cách đây
mục cha
commit
6452908067
1 tập tin đã thay đổi với 5 bổ sung1 xóa
  1. 5 1
      gym_minigrid/minigrid.py

+ 5 - 1
gym_minigrid/minigrid.py

@@ -705,6 +705,9 @@ class MiniGridEnv(gym.Env):
             'mission': StringGymSpace(min_length=0, max_length=200),
         })
 
+        # render mode
+        self.render_mode = render_mode
+
         # Range of possible rewards
         self.reward_range = (0, 1)
 
@@ -1255,7 +1258,8 @@ class MiniGridEnv(gym.Env):
         """
         Render the whole-grid human view
         """
-
+        if mode is None:
+            mode = self.render_mode
         if close:
             if self.window:
                 self.window.close()