diff --git a/flatland/envs/schedule_generators.py b/flatland/envs/schedule_generators.py index e81da924d1157fa9d5deef56306643c6a356cc8d..9b90e7c1c8d8e8c8ee09ff2412bd8fda56f57d10 100644 --- a/flatland/envs/schedule_generators.py +++ b/flatland/envs/schedule_generators.py @@ -103,7 +103,7 @@ def sparse_schedule_generator(speed_ratio_map: Mapping[float, float] = None) -> transitions = rail.get_transitions(start[0], start[1], orientation) if any(transitions) > 0: agents_direction.append(orientation) - continue + break if speed_ratio_map: speeds = speed_initialization_helper(num_agents, speed_ratio_map)