Skip to content

Commit 2135ca1

Browse files
committed
use numpy.issubdtype for a more expansive type check, was not matching int32
1 parent fb403c3 commit 2135ca1

File tree

4 files changed

+110
-152
lines changed

4 files changed

+110
-152
lines changed

rt_segment_speeds/logs/interpolate_stop_arrival.log

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -72,3 +72,4 @@ interpolate arrivals for stop_segments 2024-12-11: 2024-12-11: 0:14:09.409333
7272
2025-11-17 13:14:59.585 | INFO | interpolate_stop_arrival:interpolate_stop_arrivals:236 - interpolate arrivals for speedmap_segments 2025-11-05: 2025-11-05: 0:02:47.025833
7373
2025-11-17 15:24:22.306 | INFO | interpolate_stop_arrival:interpolate_stop_arrivals:236 - interpolate arrivals for rt_stop_times 2025-11-05: 2025-11-05: 0:10:51.471830
7474
2025-11-17 15:43:37.574 | INFO | interpolate_stop_arrival:interpolate_stop_arrivals:236 - interpolate arrivals for speedmap_segments 2025-11-05: 2025-11-05: 0:02:35.503949
75+
2026-02-02 18:13:27.813 | INFO | interpolate_stop_arrival:interpolate_stop_arrivals:181 - interpolate arrivals for rt_stop_times 2025-12-17: 2025-12-17: 0:09:55.360755

rt_segment_speeds/logs/nearest_vp.log

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -78,3 +78,4 @@
7878
2026-01-23 11:57:11.887 | INFO | nearest_vp_to_stop:nearest_neighbor_for_stop:147 - nearest neighbor for rt_stop_times 2025-12-17: 0:39:47.419228
7979
2026-02-02 16:13:13.661 | INFO | nearest_vp_to_stop:nearest_neighbor_for_stop:149 - nearest neighbor for rt_stop_times 2025-12-17: 0:41:58.668677
8080
2026-02-02 17:13:09.915 | INFO | nearest_vp_to_stop:nearest_neighbor_for_stop:149 - nearest neighbor for rt_stop_times 2025-12-17: 0:43:15.982834
81+
2026-02-02 18:03:32.371 | INFO | nearest_vp_to_stop:nearest_neighbor_for_stop:149 - nearest neighbor for rt_stop_times 2025-12-17: 0:42:34.619619

rt_segment_speeds/scripts/stop_arrivals_to_speed.py

Lines changed: 5 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -110,18 +110,11 @@ def calculate_speed_from_stop_arrivals(
110110
subseq_stop_meters=(df.groupby(trip_cols, observed=True, group_keys=False).stop_meters.shift(-1)),
111111
)
112112

113-
speed = (
114-
df.assign(
115-
meters_elapsed=df.subseq_stop_meters - df.stop_meters,
116-
sec_elapsed=df.subseq_arrival_time_sec - df.arrival_time_sec,
117-
)
118-
.pipe(
119-
segment_calcs.derive_speed,
120-
("stop_meters", "subseq_stop_meters"),
121-
("arrival_time_sec", "subseq_arrival_time_sec"),
122-
)
123-
.pipe(attach_operator_natural_identifiers, analysis_date, segment_type)
124-
)
113+
speed = df.pipe(
114+
segment_calcs.derive_speed,
115+
("stop_meters", "subseq_stop_meters"),
116+
("arrival_time_sec", "subseq_arrival_time_sec"),
117+
).pipe(attach_operator_natural_identifiers, analysis_date, segment_type)
125118

126119
gcs_pandas().data_frame_to_parquet(speed, f"{SEGMENT_GCS}{SPEED_FILE}.parquet")
127120

0 commit comments

Comments
 (0)