瀏覽代碼

Added max_steps parameter to RoomGrid environment

Maxime Chevalier-Boisvert 7 年之前
父節點
當前提交
44dedaeb16
共有 1 個文件被更改,包括 2 次插入3 次删除
  1. 2 3
      gym_minigrid/envs/roomgrid.py

+ 2 - 3
gym_minigrid/envs/roomgrid.py

@@ -44,7 +44,7 @@ class RoomGrid(MiniGridEnv):
         self,
         room_size=6,
         num_cols=4,
-        lockedRooms=False
+        max_steps=200
     ):
         assert room_size > 0
         assert room_size >= 4
@@ -52,10 +52,9 @@ class RoomGrid(MiniGridEnv):
         self.room_size = room_size
         self.num_cols = num_cols
         self.num_rows = num_cols
-        self.lockedRooms = False
 
         grid_size = (room_size - 1) * num_cols + 1
-        super().__init__(gridSize=grid_size, maxSteps=6*grid_size)
+        super().__init__(gridSize=grid_size, maxSteps=max_steps)
 
         self.reward_range = (0, 1)