diff --git a/examples/Simple_Realistic_Railway_Generator.py b/examples/Simple_Realistic_Railway_Generator.py index 82a88b5bef5f331fbf10d1ec0578a6305df1b40e..859c34a42c3da6c3bf8de03345d3659cda4ecff4 100644 --- a/examples/Simple_Realistic_Railway_Generator.py +++ b/examples/Simple_Realistic_Railway_Generator.py @@ -321,11 +321,10 @@ def realistic_rail_generator(num_cities=5, s_nodes = org_s_nodes.copy() e_nodes = org_e_nodes.copy() - for city_loop in range(len(s_nodes)): + old_cl = -1 for k in range(inter_max_number_of_connecting_tracks): sns = s_nodes[city_loop] - old_cl = -1 min_distance = np.inf end_node = None start_node = None @@ -526,9 +525,9 @@ for itrials in range(100): np.random.seed(int(time.time())) env = RailEnv(width=100, # 20+np.random.choice(100), height=100, # 20+np.random.choice(100), - rail_generator=realistic_rail_generator(num_cities=np.random.choice(10)+2, - city_size=np.random.choice(10)+10, - allowed_rotation_angles=np.arange(0,360,45), + rail_generator=realistic_rail_generator(num_cities=np.random.choice(10) + 2, + city_size=np.random.choice(10) + 10, + allowed_rotation_angles=np.arange(0, 360, 45), max_number_of_station_tracks=4, nbr_of_switches_per_station_track=2, max_number_of_connecting_tracks=4,