From 3e30bd8a9aa21ed024352fdd9040d050abb300de Mon Sep 17 00:00:00 2001
From: Erik Nygren <erik.nygren@sbb.ch>
Date: Sat, 5 Oct 2019 18:20:48 -0400
Subject: [PATCH] fixed width, height mixup
---
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 4f7e9384..d32d1521 100644
--- a/flatland/envs/rail_generators.py
+++ b/flatland/envs/rail_generators.py
@@ -558,7 +558,7 @@ def sparse_rail_generator(max_num_cities: int = 5, grid_mode: bool = False, max_
rail_trans = RailEnvTransitions()
grid_map = GridTransitionMap(width=width, height=height, transitions=rail_trans)
city_radius = int(np.ceil((max_rails_in_city + 2) / 2.0)) + 1
- vector_field = np.zeros(shape=(width, height)) - 1.
+ vector_field = np.zeros(shape=(height, width)) - 1.
min_nr_rails_in_city = 3
rails_in_city = min_nr_rails_in_city if max_rails_in_city < min_nr_rails_in_city else max_rails_in_city
--
GitLab