remove wall beneath door

epignatelli · Jun 27, 2023 · f26571f · f26571f
1 parent f2fbd61
commit f26571f
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/navix/environments/keydoor.py b/navix/environments/keydoor.py
@@ -41,6 +41,8 @@ def reset(self, key: KeyArray) -> Timestep:
         goal_pos = random_positions(k2, second_room)
         goals = Goal.create(position=goal_pos, probability=jnp.asarray(1.0))
 
+        # remove the wall beneath the door
+        grid = grid.at[tuple(door_pos)].set(0)
 
         state = State(
             key=key,