From f159b27a0bbd6fab9cf70a748c5a7c0848153133 Mon Sep 17 00:00:00 2001
From: "Egli Adrian (IT-SCI-API-PFI)" <adrian.egli@sbb.ch>
Date: Fri, 16 Aug 2019 21:00:45 +0200
Subject: [PATCH] bug fix in generator
---
flatland/envs/generators.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/flatland/envs/generators.py b/flatland/envs/generators.py
index d63e161a..272ae148 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(height - 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)
--
GitLab