diff --git a/flatland/envs/schedule_generators.py b/flatland/envs/schedule_generators.py index e81da924d1157fa9d5deef56306643c6a356cc8d..ff3cea25d70a688f7aa73ab04c90bd1ec35aea5d 100644 --- a/flatland/envs/schedule_generators.py +++ b/flatland/envs/schedule_generators.py @@ -246,3 +246,8 @@ def schedule_from_file(filename) -> ScheduleGenerator: return agents_position, agents_direction, agents_target, agents_speed, agents_malfunction return generator + + +# we can us the same schedule generator for city_rail_generator +# in order to be able to change this transparently in the future, we use a different name. +city_schedule_generator = sparse_schedule_generator