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

minor updates

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