Przeglądaj źródła

added argument agent_view_size for minigrid env (#123)

Aadesh Neupane 4 lat temu
rodzic
commit
e59b9d255f
1 zmienionych plików z 4 dodań i 2 usunięć
  1. 4 2
      gym_minigrid/roomgrid.py

+ 4 - 2
gym_minigrid/roomgrid.py

@@ -72,7 +72,8 @@ class RoomGrid(MiniGridEnv):
         num_rows=3,
         num_cols=3,
         max_steps=100,
-        seed=0
+        seed=0,
+        agent_view_size=7
     ):
         assert room_size > 0
         assert room_size >= 3
@@ -93,7 +94,8 @@ class RoomGrid(MiniGridEnv):
             height=height,
             max_steps=max_steps,
             see_through_walls=False,
-            seed=seed
+            seed=seed,
+            agent_view_size=agent_view_size
         )
 
     def room_from_pos(self, x, y):