From b1e764011dce88594bae428e21ebe7b4c53d1cd3 Mon Sep 17 00:00:00 2001
From: Erik Nygren <erik.nygren@sbb.ch>
Date: Wed, 1 May 2019 16:41:46 +0200
Subject: [PATCH] minor updates
---
flatland/core/env_observation_builder.py | 4 ++--
flatland/envs/rail_env.py | 2 ++
tests/test_rendertools.py | 4 ++--
3 files changed, 6 insertions(+), 4 deletions(-)
diff --git a/flatland/core/env_observation_builder.py b/flatland/core/env_observation_builder.py
index 1ae2819d..6c6f84eb 100644
--- a/flatland/core/env_observation_builder.py
+++ b/flatland/core/env_observation_builder.py
@@ -342,8 +342,8 @@ class TreeObsForRailEnv(ObservationBuilder):
elif num_transitions == 0:
# Wrong cell type, but let's cover it and treat it as a dead-end, just in case
- print("WRONG CELL TYPE detected in tree-search (0 transitions possible) at cell", position[0],
- position[1], direction)
+ #print("WRONG CELL TYPE detected in tree-search (0 transitions possible) at cell", position[0],
+ # position[1], direction)
last_isTerminal = True
break
diff --git a/flatland/envs/rail_env.py b/flatland/envs/rail_env.py
index f1a3d20c..7424860b 100644
--- a/flatland/envs/rail_env.py
+++ b/flatland/envs/rail_env.py
@@ -1021,6 +1021,8 @@ class RailEnv(Environment):
nbits = 0
tmp = self.rail.get_transitions((pos[0], pos[1]))
+ 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)
tmp = tmp >> 1
diff --git a/tests/test_rendertools.py b/tests/test_rendertools.py
index 1f5c3179..73b42c38 100644
--- a/tests/test_rendertools.py
+++ b/tests/test_rendertools.py
@@ -30,8 +30,8 @@ def checkFrozenImage(sFileImage):
if bytesFrozenImage is None:
bytesFrozenImage = bytesImage
else:
- assert(bytesFrozenImage.shape == bytesImage.shape)
- assert((np.sum(np.square(bytesFrozenImage - bytesImage)) / bytesFrozenImage.size) < 1e-3)
+ assert (bytesFrozenImage.shape == bytesImage.shape)
+ assert ((np.sum(np.square(bytesFrozenImage - bytesImage)) / bytesFrozenImage.size) < 1e-3)
def test_render_env():
--
GitLab