test adding a column to python model

cal-itp · Mar 28, 2023 · a07e455 · a07e455
1 parent 38ebd78
commit a07e455
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/warehouse/models/intermediate/gtfs/int_gtfs_rt__rt_segments.py b/warehouse/models/intermediate/gtfs/int_gtfs_rt__rt_segments.py
@@ -1,7 +1,7 @@
 def model(dbt, session):
     df = (
         dbt.ref("fct_vehicle_locations")
-        .select("key", "vehicle_trip_key", "location")
+        .select("key", "vehicle_trip_key", "location", "next_location_key")
         .where("dt = '2023-03-01'")
     )
     return df