瀏覽代碼

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

saleml 2 年之前
父節點
當前提交
6452908067
共有 1 個文件被更改,包括 5 次插入1 次删除
  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()