From 7d8f19b807c3b1255920021051fcb03f0aa7f13b Mon Sep 17 00:00:00 2001
From: Erik Nygren <erik.nygren@sbb.ch>
Date: Sat, 31 Aug 2019 20:12:08 -0400
Subject: [PATCH] minor bugfix in level generator
---
flatland/envs/rail_generators.py | 3 ++-
1 file changed, 2 insertions(+), 1 deletion(-)
diff --git a/flatland/envs/rail_generators.py b/flatland/envs/rail_generators.py
index 1b72b978..c100e3db 100644
--- a/flatland/envs/rail_generators.py
+++ b/flatland/envs/rail_generators.py
@@ -714,7 +714,8 @@ def sparse_rail_generator(num_cities=5, num_intersections=4, num_trainstations=2
(station_x, station_y))
# Check if connection was made
if len(connection) == 0:
- train_stations[trainstation_node].pop(-1)
+ if len(train_stations) > 0:
+ train_stations[trainstation_node].pop(-1)
else:
built_num_trainstation += 1
--
GitLab