|
@@ -20,7 +20,7 @@ class ObstructedMazeEnv(RoomGrid):
|
|
|
max_steps=max_steps,
|
|
|
**kwargs
|
|
|
)
|
|
|
- self.obj = Ball() # intiale the obj attribute, that will be changed later on
|
|
|
+ self.obj = Ball() # initialize the obj attribute, that will be changed later on
|
|
|
|
|
|
def _gen_grid(self, width, height):
|
|
|
super()._gen_grid(width, height)
|
|
@@ -67,7 +67,8 @@ class ObstructedMazeEnv(RoomGrid):
|
|
|
if blocked:
|
|
|
vec = DIR_TO_VEC[door_idx]
|
|
|
blocking_ball = Ball(self.blocking_ball_color) if blocked else None
|
|
|
- self.grid.set(door_pos[0] - vec[0], door_pos[1] - vec[1], blocking_ball)
|
|
|
+ self.grid.set(door_pos[0] - vec[0],
|
|
|
+ door_pos[1] - vec[1], blocking_ball)
|
|
|
|
|
|
if locked:
|
|
|
obj = Key(door.color)
|
|
@@ -105,7 +106,8 @@ class ObstructedMaze_1Dlhb(ObstructedMazeEnv):
|
|
|
blocked=self.blocked,
|
|
|
)
|
|
|
|
|
|
- self.obj, _ = self.add_object(1, 0, "ball", color=self.ball_to_find_color)
|
|
|
+ self.obj, _ = self.add_object(
|
|
|
+ 1, 0, "ball", color=self.ball_to_find_color)
|
|
|
self.place_agent(0, 0)
|
|
|
|
|
|
|