From f26571f0d8800b3d971943493bda6865189e570d Mon Sep 17 00:00:00 2001
From: epignatelli <edu.pignatelli@gmail.com>
Date: Tue, 27 Jun 2023 12:28:25 +0100
Subject: [PATCH] remove wall beneath door

---
 navix/environments/keydoor.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/navix/environments/keydoor.py b/navix/environments/keydoor.py
index e414aa6..fa1b760 100644
--- 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,