From 3528da79555b9dd2d5827f6e88412e63cea6fc25 Mon Sep 17 00:00:00 2001
From: "Egli Adrian (IT-SCI-API-PFI)" <adrian.egli@sbb.ch>
Date: Fri, 16 Aug 2019 21:02:59 +0200
Subject: [PATCH] bug fix in generator

---
 flatland/envs/generators.py                      | 2 +-
 tests/test_flatland_env_sparse_rail_generator.py | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/flatland/envs/generators.py b/flatland/envs/generators.py
index 272ae148..aef00cdd 100644
--- a/flatland/envs/generators.py
+++ b/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)
diff --git a/tests/test_flatland_env_sparse_rail_generator.py b/tests/test_flatland_env_sparse_rail_generator.py
index 49739297..3c1bc18b 100644
--- a/tests/test_flatland_env_sparse_rail_generator.py
+++ b/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),
-- 
GitLab