From c81fd51b2bc8fa42857247fb2121eed7e2cad8af Mon Sep 17 00:00:00 2001
From: SP Mohanty <spmohanty91@gmail.com>
Date: Mon, 29 Apr 2019 15:01:56 +0200
Subject: [PATCH] Fix linter errors
---
flatland/core/env_observation_builder.py | 2 +-
flatland/envs/rail_env.py | 2 +-
2 files changed, 2 insertions(+), 2 deletions(-)
diff --git a/flatland/core/env_observation_builder.py b/flatland/core/env_observation_builder.py
index 848b184f..ac5bfd8b 100644
--- a/flatland/core/env_observation_builder.py
+++ b/flatland/core/env_observation_builder.py
@@ -242,7 +242,7 @@ class TreeObsForRailEnv(ObservationBuilder):
# Start from the current orientation, and see which transitions are available;
# 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)]:
- #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):
new_cell = self._new_position(position, branch_direction)
diff --git a/flatland/envs/rail_env.py b/flatland/envs/rail_env.py
index 30e474d6..01562a7f 100644
--- a/flatland/envs/rail_env.py
+++ b/flatland/envs/rail_env.py
@@ -905,7 +905,7 @@ class RailEnv(Environment):
def reset(self, regen_rail=True, replace_agents=True):
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.fill_valid_positions()
--
GitLab