diff --git a/flatland/envs/rail_generators_city_generator.py b/flatland/envs/rail_generators_city_generator.py index b4d854bbde0ef3d5b1ed810ef809b8514187a8ab..36b8b84638c300237b0dd528ee50f3c0a878d42e 100644 --- a/flatland/envs/rail_generators_city_generator.py +++ b/flatland/envs/rail_generators_city_generator.py @@ -416,8 +416,6 @@ def city_generator(num_cities: int = 5, if print_out_info: print("intern_connect_max_nbr_of_shortes_city:", intern_connect_max_nbr_of_shortes_city) - agent_start_targets_nodes = [] - # ---------------------------------------------------------------------------------- # generate city locations generate_city_locations, max_num_cities = do_generate_city_locations(width, height, intern_city_size, @@ -465,6 +463,7 @@ def city_generator(num_cities: int = 5, node_available_target.append(len(train_stations_slots[node_idx])) # Assign agents to slots + agent_start_targets_nodes = [] for agent_idx in range(num_agents): avail_start_nodes = [idx for idx, val in enumerate(node_available_start) if val > 0] avail_target_nodes = [idx for idx, val in enumerate(node_available_target) if val > 0]