fix exploration bug

bruehldev · Jan 20, 2019 · a70cb40 · a70cb40
1 parent 2b41f0c
commit a70cb40
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/babyai/bot.py b/babyai/bot.py
@@ -752,7 +752,8 @@ def take_action(self, action):
 
         # Find the closest unseen position
         _, unseen_pos, with_blockers = self.bot.shortest_path(
-            lambda pos, cell: not self.bot.vis_mask[pos]
+            lambda pos, cell: not self.bot.vis_mask[pos],
+            try_with_blockers=True
         )
 
         if unseen_pos: