Parcourir la source

Added goal_pos variable to MultiRoomEnv

Maxime Chevalier-Boisvert il y a 7 ans
Parent
commit
d5d117bb78
1 fichiers modifiés avec 1 ajouts et 1 suppressions
  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'