bug fix in generator

flatland/envs/generators.py

@@ -733,7 +733,7 @@ def realistic_rail_generator(nr_start_goal=1,  seed=0):
                              (x_offsets[off_set_loop] - 1, 0),
                              (x_offsets[off_set_loop] - 2, 0))
 
-            for nbr_track_loop in range(max_n_track_seg-1): 
+            for nbr_track_loop in range(max_n_track_seg-1):
                 if len(data) < 2 * n_track_seg + 1:
                     break
                 x = np.sort(np.random.choice(data, 2 * n_track_seg, False)).astype(int)

tests/test_flatland_env_sparse_rail_generator.py

@@ -8,7 +8,7 @@ from flatland.utils.rendertools import RenderTool
 import numpy as np
 
 def test_realistic_rail_generator():
-    for test_loop in range(5):
+    for test_loop in range(20):
         num_agents = np.random.randint(10,30)
         env = RailEnv(width=np.random.randint(40,80),
                       height=np.random.randint(10,20),