From cf1cc9b79855ed18444c1d04b18e53a7e8d70c70 Mon Sep 17 00:00:00 2001
From: hagrid67 <jdhwatson@gmail.com>
Date: Wed, 1 May 2019 19:51:48 +0100
Subject: [PATCH] fixed more lint
---
flatland/envs/rail_env.py | 7 +++++--
1 file changed, 5 insertions(+), 2 deletions(-)
diff --git a/flatland/envs/rail_env.py b/flatland/envs/rail_env.py
index a236362d..8969af3e 100644
--- a/flatland/envs/rail_env.py
+++ b/flatland/envs/rail_env.py
@@ -1041,7 +1041,10 @@ class RailEnv(Environment):
nbits = 0
tmp = self.rail.get_transitions((pos[0], pos[1]))
possible_transitions = self.rail.get_transitions((pos[0], pos[1], direction))
- # print(np.sum(self.rail.get_transitions((pos[0], pos[1],direction))),self.rail.get_transitions((pos[0], pos[1],direction)),self.rail.get_transitions((pos[0], pos[1])),(pos[0], pos[1],direction))
+ # print(np.sum(self.rail.get_transitions((pos[0], pos[1],direction))),
+ # self.rail.get_transitions((pos[0], pos[1],direction)),
+ # self.rail.get_transitions((pos[0], pos[1])),
+ # (pos[0], pos[1],direction))
while tmp > 0:
nbits += (tmp & 1)
@@ -1087,7 +1090,7 @@ class RailEnv(Environment):
if np.sum(possible_transitions) == 1:
# Checking for curves
curv_dir = np.argmax(possible_transitions)
- #valid_transition = self.rail.get_transition(
+ # valid_transition = self.rail.get_transition(
# (pos[0], pos[1], direction),
# movement)
movement = curv_dir
--
GitLab