From b216dc9460a72f72835fe97998acd88c18371454 Mon Sep 17 00:00:00 2001
From: "Egli Adrian (IT-SCI-API-PFI)" <adrian.egli@sbb.ch>
Date: Mon, 19 Aug 2019 09:06:50 +0200
Subject: [PATCH] bug fixed
---
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 de34246d..6dff5d95 100644
--- a/flatland/envs/generators.py
+++ b/flatland/envs/generators.py
@@ -779,7 +779,7 @@ def realistic_rail_generator(nr_start_goal=1, seed=0,max_add_dead_end = 7):
# add dead-end
if True:
- n = int(np.random.choice(np.arange(max_add_dead_end), 1)[0])
+ n = max_add_dead_end#int(np.random.choice(np.arange(max_add_dead_end-2)+1, 1)[0])
for pos_y in np.random.choice(np.arange(width - 7) + 3, n):
pos_x = off_set
pos_x1 = max(0, min(pos_x + 1, height - 1))
--
GitLab