Explorar o código

the doors should also have a position (#46)

Dzmitry Bahdanau %!s(int64=6) %!d(string=hai) anos
pai
achega
a256e6a6db
Modificáronse 1 ficheiros con 1 adicións e 0 borrados
  1. 1 0
      gym_minigrid/roomgrid.py

+ 1 - 0
gym_minigrid/roomgrid.py

@@ -237,6 +237,7 @@ class RoomGrid(MiniGridEnv):
 
         pos = room.door_pos[door_idx]
         self.grid.set(*pos, door)
+        door.cur_pos = pos
 
         neighbor = room.neighbors[door_idx]
         room.doors[door_idx] = door