fix pre_route bug

tsinghua-fib-lab · Sep 24, 2024 · c2e0473 · c2e0473
1 parent 0604c9a
commit c2e0473
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/mosstool/trip/route/preroute.py b/mosstool/trip/route/preroute.py
@@ -41,6 +41,7 @@ async def pre_route(
  departure_time = None
  all_schedules = list(person.schedules)
  person.ClearField("schedules")
+ good_schedules = []
  for schedule in all_schedules:
  schedule = cast(Schedule, schedule)
  if schedule.HasField("departure_time"):
@@ -98,4 +99,6 @@ async def pre_route(
  good_schedule.CopyFrom(schedule)
  good_schedule.ClearField("trips")
  good_schedule.trips.extend(good_trips)
+ good_schedules.append(good_schedule)
+ person.schedules.extend(good_schedules)
  return person