Skip to content
Snippets Groups Projects
Commit 99c7f60b authored by Erik Nygren's avatar Erik Nygren
Browse files

fixed minimal city size

parent be00b850
No related branches found
No related tags found
No related merge requests found
...@@ -37,7 +37,7 @@ env = RailEnv(width=50, ...@@ -37,7 +37,7 @@ env = RailEnv(width=50,
seed=1, # Random seed seed=1, # Random seed
grid_mode=False, grid_mode=False,
max_rails_between_cities=3, max_rails_between_cities=3,
max_rails_in_city=6, max_rails_in_city=2,
), ),
schedule_generator=sparse_schedule_generator(speed_ration_map), schedule_generator=sparse_schedule_generator(speed_ration_map),
number_of_agents=10, number_of_agents=10,
......
...@@ -558,7 +558,7 @@ def sparse_rail_generator(max_num_cities: int = 5, grid_mode: bool = False, max_ ...@@ -558,7 +558,7 @@ def sparse_rail_generator(max_num_cities: int = 5, grid_mode: bool = False, max_
rail_trans = RailEnvTransitions() rail_trans = RailEnvTransitions()
grid_map = GridTransitionMap(width=width, height=height, transitions=rail_trans) 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. vector_field = np.zeros(shape=(height, width)) - 1.
min_nr_rails_in_city = 3 min_nr_rails_in_city = 3
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment