Quellcode durchsuchen

Added goal_pos variable to MultiRoomEnv

Maxime Chevalier-Boisvert vor 7 Jahren
Ursprung
Commit
d5d117bb78
1 geänderte Dateien mit 1 neuen und 1 gelöschten Zeilen
  1. 1 1
      gym_minigrid/envs/multiroom.py

+ 1 - 1
gym_minigrid/envs/multiroom.py

@@ -112,7 +112,7 @@ class MultiRoomEnv(MiniGridEnv):
         self.place_agent(roomList[0].top, roomList[0].size)
 
         # Place the final goal in the last room
-        self.place_obj(Goal(), roomList[-1].top, roomList[-1].size)
+        self.goal_pos = self.place_obj(Goal(), roomList[-1].top, roomList[-1].size)
 
         self.mission = 'traverse the rooms to get to the goal'