From f04879914cba844829df958a6213d1140b105c40 Mon Sep 17 00:00:00 2001
From: "Egli Adrian (IT-SCI-API-PFI)" <adrian.egli@sbb.ch>
Date: Fri, 23 Aug 2019 11:18:11 +0200
Subject: [PATCH] .
---
flatland/envs/generators.py | 4 ++++
1 file changed, 4 insertions(+)
diff --git a/flatland/envs/generators.py b/flatland/envs/generators.py
index 127e8790..bfcfdf71 100644
--- a/flatland/envs/generators.py
+++ b/flatland/envs/generators.py
@@ -923,6 +923,10 @@ def realistic_rail_generator(nr_start_goal=1, seed=0, add_max_dead_end=4, goals_
agents_targets.pop(t)
agents_target.append((tp[0], tp[1]))
+ if len(agents_position) == 0:
+ print("no more position left")
+ break
+
sel = np.random.choice(range(len(agents_positions)))
# backward
p = agents_positions[sel]
--
GitLab