diff --git a/examples/Simple_Realistic_Railway_Generator.py b/examples/Simple_Realistic_Railway_Generator.py index c231971b3c443a09ff8d6718ef00da5459dc0794..2057cbc1dd986d4e23df9f3d4cfeeedd91a35b6a 100644 --- a/examples/Simple_Realistic_Railway_Generator.py +++ b/examples/Simple_Realistic_Railway_Generator.py @@ -262,7 +262,7 @@ def realistic_rail_generator(num_cities=5, station_slots[city_loop].append(connection[int(np.floor(len(connection) / 2))]) nodes_added.append(start_node) nodes_added.append(end_node) - + if print_out_info: print("max nbr of station slots with given configuration is:", station_slots_cnt) @@ -441,15 +441,15 @@ def realistic_rail_generator(num_cities=5, for itrials in range(100): print(itrials, "generate new city") np.random.seed(int(time.time())) - env = RailEnv(width=120, - height=120, - rail_generator=realistic_rail_generator(num_cities=10, - city_size=20, - allowed_rotation_angles=[-90,0,90], + env = RailEnv(width=20+np.random.choice(100), + height=20+np.random.choice(100), + rail_generator=realistic_rail_generator(num_cities=2+np.random.choice(10), + city_size=4+np.random.choice(20), + allowed_rotation_angles=[-90,-30,0,30,90], max_number_of_station_tracks=4, nbr_of_switches_per_station_track=2, - max_number_of_connecting_tracks=1, - do_random_connect_stations=True, + max_number_of_connecting_tracks=3, + do_random_connect_stations=False, # Number of cities in map seed=int(time.time()) # Random seed ),