minor updates

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
 

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

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():