|
@@ -16,7 +16,7 @@ class DistShiftEnv(MiniGridEnv):
|
|
):
|
|
):
|
|
self.agent_start_pos = agent_start_pos
|
|
self.agent_start_pos = agent_start_pos
|
|
self.agent_start_dir = agent_start_dir
|
|
self.agent_start_dir = agent_start_dir
|
|
- self.goal_pos=(width-2, 1)
|
|
|
|
|
|
+ self.goal_pos = (width-2, 1)
|
|
self.strip2_row = strip2_row
|
|
self.strip2_row = strip2_row
|
|
|
|
|
|
super().__init__(
|
|
super().__init__(
|