From c8ed47160c5c563d0d5dba68a1792425bd56b911 Mon Sep 17 00:00:00 2001
From: "Egli Adrian (IT-SCI-API-PFI)" <adrian.egli@sbb.ch>
Date: Tue, 10 Sep 2019 18:06:53 +0200
Subject: [PATCH] check fixed
---
flatland/envs/rail_generators.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/flatland/envs/rail_generators.py b/flatland/envs/rail_generators.py
index d6e1c34f..a16fb601 100644
--- a/flatland/envs/rail_generators.py
+++ b/flatland/envs/rail_generators.py
@@ -666,7 +666,7 @@ def sparse_rail_generator(num_cities=5, num_intersections=4, num_trainstations=2
0,
width - 1)
tries = 0
- while (station_x, station_y) in train_stations \
+ while (station_x, station_y) in train_stations[trainstation_node] \
or (station_x, station_y) == node_positions[trainstation_node] \
or rail_array[(station_x, station_y)] != 0: # noqa: E125
--
GitLab