Commit 1d7fb88f authored by jameskrw's avatar jameskrw
Browse files

updated env states

parent 1b22f34c
Loading
Loading
Loading
Loading
+2 −1
Original line number Diff line number Diff line
@@ -205,7 +205,8 @@ class SokobanEnv(BaseEnv):
            "player_position": player_pos,
            "box_positions": box_positions,
            "target_positions": target_positions,
            "wall_positions": wall_positions
            "wall_positions": wall_positions,
            "grid_size":room_state.shape
        }