From e1332b10bf5773f4d92805c5ed559b9a596a2f7d Mon Sep 17 00:00:00 2001
From: "Egli Adrian (IT-SCI-API-PFI)" <adrian.egli@sbb.ch>
Date: Tue, 20 Aug 2019 17:48:11 +0200
Subject: [PATCH] bug fix
---
flatland/envs/generators.py | 13 ++++++++++++-
1 file changed, 12 insertions(+), 1 deletion(-)
diff --git a/flatland/envs/generators.py b/flatland/envs/generators.py
index bf9679f9..8fa40c8e 100644
--- a/flatland/envs/generators.py
+++ b/flatland/envs/generators.py
@@ -896,6 +896,17 @@ def realistic_rail_generator(nr_start_goal=1, seed=0, add_max_dead_end=4, two_tr
agents_target = []
agents_direction = []
+ for a in range(len(agents_positions)):
+ cell_transitions = grid_map.get_transitions(agents_positions[a][0], agents_positions[a][1],
+ agents_directions[a])
+ if np.sum(cell_transitions) == 0:
+ agents_directions[a] += 2
+ agents_directions[a] = agents_directions[a] % 4
+ cell_transitions = grid_map.get_transitions(agents_positions[a][0], agents_positions[a][1],
+ agents_directions[a])
+ if np.sum(cell_transitions):
+ print(a, "bug")
+
for a in range(min(len(agents_targets), num_agents)):
t = np.random.choice(range(len(agents_targets)))
d = agents_targets[t]
@@ -1047,7 +1058,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