diff --git a/examples/flatland_2_0_example.py b/examples/flatland_2_0_example.py index 4a614b2ff303ae17e6f739a078c1da34efc15185..2d393fd43db84042760b2b1bf9f0455b9d53d7b2 100644 --- a/examples/flatland_2_0_example.py +++ b/examples/flatland_2_0_example.py @@ -34,7 +34,7 @@ env = RailEnv(width=50, seed=0, # Random seed grid_mode=False, max_inter_city_rails=2, - max_tracks_in_city=8, + max_tracks_in_city=4, ), schedule_generator=sparse_schedule_generator(), number_of_agents=10, diff --git a/flatland/envs/rail_generators.py b/flatland/envs/rail_generators.py index 74d17b837ca53367c93ff938df684f878b479cd2..725943a49555dcdb6695952748f9393cbb619665 100644 --- a/flatland/envs/rail_generators.py +++ b/flatland/envs/rail_generators.py @@ -551,7 +551,7 @@ def sparse_rail_generator(num_cities=5, grid_mode=False, max_inter_city_rails=4, rail_array = grid_map.grid rail_array.fill(0) np.random.seed(seed + num_resets) - node_radius = int(np.ceil(max_tracks_in_city / 2)) + node_radius = int(np.ceil(max_tracks_in_city / 2)) + 2 max_inter_city_rails_allowed = max_inter_city_rails if max_inter_city_rails_allowed > max_tracks_in_city: max_inter_city_rails_allowed = max_tracks_in_city @@ -616,8 +616,8 @@ def sparse_rail_generator(num_cities=5, grid_mode=False, max_inter_city_rails=4, tries = 0 while to_close: - x_tmp = node_radius + 1 + np.random.randint(height - 2 * node_radius) - y_tmp = node_radius + 1 + np.random.randint(width - 2 * node_radius) + x_tmp = node_radius + 1 + np.random.randint(height - 2 * node_radius - 1) + y_tmp = node_radius + 1 + np.random.randint(width - 2 * node_radius - 1) to_close = False # Check distance to nodes for node_pos in node_positions: @@ -806,6 +806,7 @@ def sparse_rail_generator(num_cities=5, grid_mode=False, max_inter_city_rails=4, if 1 <= nbits <= 2: built_num_trainstations += 1 train_stations[current_city].append(possible_location) + print(built_num_trainstations) return train_stations, built_num_trainstations def _generate_start_target_pairs(num_agents, nb_nodes, train_stations):