From eb0cb675324d7f94b719a4749f173c1b86c07e4d Mon Sep 17 00:00:00 2001
From: "Egli Adrian (IT-SCI-API-PFI)" <adrian.egli@sbb.ch>
Date: Tue, 20 Aug 2019 16:38:33 +0200
Subject: [PATCH] flake8 fixes, and realistic generator fix
---
flatland/envs/generators.py | 7 +++----
1 file changed, 3 insertions(+), 4 deletions(-)
diff --git a/flatland/envs/generators.py b/flatland/envs/generators.py
index 808833c9..bf9679f9 100644
--- a/flatland/envs/generators.py
+++ b/flatland/envs/generators.py
@@ -680,8 +680,8 @@ def realistic_rail_generator(nr_start_goal=1, seed=0, add_max_dead_end=4, two_tr
for off_set_loop in range(len(x_offsets)):
off_set = x_offsets[off_set_loop]
# second track
- data = np.arange(2, width - 2)
- n_track_seg = np.random.choice([1, 2])
+ data = np.arange(4, width - 4)
+ n_track_seg = np.random.choice([1, 2, 3])
track_2 = False
if two_track_back_bone:
@@ -1047,8 +1047,7 @@ def sparse_rail_generator(num_cities=100, num_intersections=10, num_trainstation
0,
width - 1)
while (station_x, station_y) in train_stations or (station_x, station_y) == node_positions[
- trainstation_node] or \
- rail_array[(station_x, station_y)] != 0:
+ trainstation_node] or rail_array[(station_x, station_y)] != 0:
station_x = np.clip(
node_positions[trainstation_node][0] + np.random.randint(-node_radius, node_radius),
0,
--
GitLab