diff --git a/flatland/envs/schedule_generators.py b/flatland/envs/schedule_generators.py index ff3cea25d70a688f7aa73ab04c90bd1ec35aea5d..ba703acf5e35e502d0c6248571a9a795a1676352 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)