From 8bbf5b62f603655c5783dcecb05f77a311bff324 Mon Sep 17 00:00:00 2001
From: MLErik <baerenjesus@gmail.com>
Date: Fri, 23 Aug 2019 18:05:16 -0400
Subject: [PATCH] stability improvements
---
flatland/envs/generators.py | 5 ++++-
1 file changed, 4 insertions(+), 1 deletion(-)
diff --git a/flatland/envs/generators.py b/flatland/envs/generators.py
index f79c5600..e3c978e1 100644
--- a/flatland/envs/generators.py
+++ b/flatland/envs/generators.py
@@ -1097,6 +1097,7 @@ def sparse_rail_generator(num_cities=5, num_intersections=4, num_trainstations=2
if num_cities > 1:
train_stations = [[] for i in range(num_cities)]
built_num_trainstation = 0
+ spot_found = True
for station in range(num_trainstations):
trainstation_node = int(station / num_trainstations * num_cities)
@@ -1120,8 +1121,10 @@ def sparse_rail_generator(num_cities=5, num_intersections=4, num_trainstations=2
tries += 1
if tries > 100:
warnings.warn("Could not set trainstations, please change initial parameters!!!!")
+ spot_found = False
break
- train_stations[trainstation_node].append((station_x, station_y))
+ if spot_found:
+ train_stations[trainstation_node].append((station_x, station_y))
# Connect train station to the correct node
connection = connect_from_nodes(rail_trans, rail_array, node_positions[trainstation_node],
--
GitLab