diff --git a/examples/Simple_Realistic_Railway_Generator.py b/examples/Simple_Realistic_Railway_Generator.py index 98a2e39e778a69b78c51cef664d443c859feaf34..f1649e7814769529b4167efb089e88b27c1ecf81 100644 --- a/examples/Simple_Realistic_Railway_Generator.py +++ b/examples/Simple_Realistic_Railway_Generator.py @@ -468,19 +468,19 @@ for itrials in range(1000): np.random.seed(0 * int(time.time())) env = RailEnv(width=40 + np.random.choice(100), height=40 + np.random.choice(100), - rail_generator=realistic_rail_generator(num_cities=2000 + np.random.choice(10), + rail_generator=realistic_rail_generator(num_cities=2 + np.random.choice(10), city_size=10 + np.random.choice(10), allowed_rotation_angles=np.arange(-180, 180, 15), max_number_of_station_tracks=np.random.choice(4) + 4, nbr_of_switches_per_station_track=np.random.choice(4) + 2, - connect_max_nbr_of_shortes_city=np.random.choice(4) + 2, - do_random_connect_stations=np.random.choice(1) == 0, + connect_max_nbr_of_shortes_city=2, + do_random_connect_stations=False and np.random.choice(1) == 0, # Number of cities in map seed=int(time.time()), # Random seed print_out_info=False ), schedule_generator=sparse_schedule_generator(), - number_of_agents=1, + number_of_agents=100, obs_builder_object=GlobalObsForRailEnv()) # reset to initialize agents_static @@ -497,5 +497,4 @@ for itrials in range(1000): "flatland_frame_{:04d}_{:04d}.png".format(itrials, 0) )) - input() env_renderer.close_window()