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