소스 검색

Added goal_pos variable to MultiRoomEnv

Maxime Chevalier-Boisvert 7 년 전
부모
커밋
d5d117bb78
1개의 변경된 파일1개의 추가작업 그리고 1개의 파일을 삭제
  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'