Commit c81fd51b authored by spmohanty's avatar spmohanty
Browse files

Fix linter errors

parent e71f0572
Pipeline #429 passed with stage
in 2 minutes and 41 seconds
...@@ -242,7 +242,7 @@ class TreeObsForRailEnv(ObservationBuilder): ...@@ -242,7 +242,7 @@ class TreeObsForRailEnv(ObservationBuilder):
# Start from the current orientation, and see which transitions are available; # Start from the current orientation, and see which transitions are available;
# organize them as [left, forward, right, back], relative to the current orientation # organize them as [left, forward, right, back], relative to the current orientation
for branch_direction in [(orientation + 4 + i) % 4 for i in range(-1, 3)]: for branch_direction in [(orientation + 4 + i) % 4 for i in range(-1, 3)]:
#TODO: check if cell is a curve, then keep branch direction forward instead of left or right # TODO: check if cell is a curve, then keep branch direction forward instead of left or right
if self.env.rail.get_transition((position[0], position[1], orientation), branch_direction): if self.env.rail.get_transition((position[0], position[1], orientation), branch_direction):
new_cell = self._new_position(position, branch_direction) new_cell = self._new_position(position, branch_direction)
......
...@@ -905,7 +905,7 @@ class RailEnv(Environment): ...@@ -905,7 +905,7 @@ class RailEnv(Environment):
def reset(self, regen_rail=True, replace_agents=True): def reset(self, regen_rail=True, replace_agents=True):
if regen_rail or self.rail is None: if regen_rail or self.rail is None:
#TODO: Import not only rail information but also start and goal positions # TODO: Import not only rail information but also start and goal positions
self.rail = self.rail_generator(self.width, self.height, self.num_resets) self.rail = self.rail_generator(self.width, self.height, self.num_resets)
self.fill_valid_positions() self.fill_valid_positions()
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment