From a62da53e98504ae186c3c43a947dd34fb0af4881 Mon Sep 17 00:00:00 2001
From: "Egli Adrian (IT-SCI-API-PFI)" <adrian.egli@sbb.ch>
Date: Tue, 10 Sep 2019 18:05:21 +0200
Subject: [PATCH] bug 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 5af4a079..d6e1c34f 100644
--- a/flatland/envs/rail_generators.py
+++ b/flatland/envs/rail_generators.py
@@ -572,7 +572,7 @@ def sparse_rail_generator(num_cities=5, num_intersections=4, num_trainstations=2
nodes_per_col = int(np.ceil(nb_nodes / nodes_per_row))
x_positions = np.linspace(node_radius, height - node_radius, nodes_per_row, dtype=int)
y_positions = np.linspace(node_radius, width - node_radius, nodes_per_col, dtype=int)
- city_idx = np.random.choice(np.arange(nb_nodes), num_cities)
+ city_idx = np.random.choice(np.arange(nb_nodes), num_cities, False)
node_positions = _generate_node_positions_grid_mode(city_idx, city_positions, intersection_positions,
nb_nodes,
--
GitLab