Merge pull request mila-iqia#35 from mila-udem/maximecb-botgridfix

Updated bot following MiniGrid change
mattdeitke · Jan 29, 2019 · 95411ff · 95411ff
2 parents 58d3325 + f0d5038
commit 95411ff
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/babyai/bot.py b/babyai/bot.py
@@ -501,13 +501,11 @@ def __init__(self, mission):
         # Mission to be solved
         self.mission = mission
 
-        grid_size = mission.grid_size
-
         # Grid containing what has been mapped out
-        self.grid = Grid(grid_size, grid_size)
+        self.grid = Grid(mission.width, mission.height)
 
         # Visibility mask. True for explored/seen, false for unexplored.
-        self.vis_mask = np.zeros(shape=(grid_size, grid_size), dtype=np.bool)
+        self.vis_mask = np.zeros(shape=(mission.width, mission.height), dtype=np.bool)
 
         # Stack of tasks/subtasks to complete (tuples)
         self.stack = []