add run number check

Intersection · Nov 18, 2024 · 9530fb5 · 9530fb5
1 parent 1fef0d5
commit 9530fb5
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/gtfs_realtime_translators/factories/factories.py b/gtfs_realtime_translators/factories/factories.py
@@ -72,7 +72,7 @@ def create(*args, **kwargs):
                                                                    departure=departure,
                                                                    stop_id=stop_id)
 
-        vehicle_descriptor = gtfs_realtime.VehicleDescriptor(run_number=run_number)
+        vehicle_descriptor = gtfs_realtime.VehicleDescriptor()
 
         if track:
             stop_time_update.Extensions[intersection_gtfs_realtime.intersection_stop_time_update].track = track
@@ -82,8 +82,8 @@ def create(*args, **kwargs):
             stop_time_update.Extensions[intersection_gtfs_realtime.intersection_stop_time_update].scheduled_departure.time = scheduled_departure
         if stop_name:
             stop_time_update.Extensions[intersection_gtfs_realtime.intersection_stop_time_update].stop_name = stop_name
-        # if run_number:
-        #     vehicle_descriptor.Extensions[intersection_gtfs_realtime.intersection_vehicle_descriptor].run_number = run_number
+        if run_number:
+            vehicle_descriptor.Extensions[intersection_gtfs_realtime.intersection_vehicle_descriptor].run_number = run_number
 
         trip_update = gtfs_realtime.TripUpdate(trip=trip_descriptor,
                                                stop_time_update=[stop_time_update],