From 99c7f60b9efa59624368edadc0b4d75bd7e00abc Mon Sep 17 00:00:00 2001
From: Erik Nygren <erik.nygren@sbb.ch>
Date: Sat, 5 Oct 2019 23:46:52 -0400
Subject: [PATCH] fixed minimal city size
---
examples/flatland_2_0_example.py | 2 +-
flatland/envs/rail_generators.py | 2 +-
2 files changed, 2 insertions(+), 2 deletions(-)
diff --git a/examples/flatland_2_0_example.py b/examples/flatland_2_0_example.py
index f3bfef28..a5a5946b 100644
--- a/examples/flatland_2_0_example.py
+++ b/examples/flatland_2_0_example.py
@@ -37,7 +37,7 @@ env = RailEnv(width=50,
seed=1, # Random seed
grid_mode=False,
max_rails_between_cities=3,
- max_rails_in_city=6,
+ max_rails_in_city=2,
),
schedule_generator=sparse_schedule_generator(speed_ration_map),
number_of_agents=10,
diff --git a/flatland/envs/rail_generators.py b/flatland/envs/rail_generators.py
index 324e21b6..2f974ae9 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
+ city_radius = int(np.ceil((max_rails_in_city + 2) / 2.0)) + 2
vector_field = np.zeros(shape=(height, width)) - 1.
min_nr_rails_in_city = 3
--
GitLab