Skip to content
Snippets Groups Projects
Commit 391ca4a3 authored by Erik Nygren's avatar Erik Nygren :bullettrain_front:
Browse files

squashed bug where agent was placed in target city

parent b3657480
No related branches found
No related tags found
No related merge requests found
...@@ -28,16 +28,16 @@ speed_ration_map = {1.: 0.25, # Fast passenger train ...@@ -28,16 +28,16 @@ speed_ration_map = {1.: 0.25, # Fast passenger train
1. / 3.: 0.25, # Slow commuter train 1. / 3.: 0.25, # Slow commuter train
1. / 4.: 0.25} # Slow freight train 1. / 4.: 0.25} # Slow freight train
env = RailEnv(width=75, env = RailEnv(width=100,
height=75, height=20,
rail_generator=sparse_rail_generator(num_cities=15, # Number of cities in map (where train stations are) rail_generator=sparse_rail_generator(num_cities=2, # Number of cities in map (where train stations are)
seed=1, # Random seed seed=1, # Random seed
grid_mode=False, grid_mode=True,
max_inter_city_rails=2, max_inter_city_rails=2,
max_tracks_in_city=8, max_tracks_in_city=8,
), ),
schedule_generator=sparse_schedule_generator(speed_ration_map), schedule_generator=sparse_schedule_generator(speed_ration_map),
number_of_agents=200, number_of_agents=20,
stochastic_data=stochastic_data, # Malfunction data generator stochastic_data=stochastic_data, # Malfunction data generator
obs_builder_object=GlobalObsForRailEnv(), obs_builder_object=GlobalObsForRailEnv(),
remove_agents_at_target=True remove_agents_at_target=True
......
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