fix: fix virtual stop selection if MAX_STOPS_PER_TRIP is exceeded

mfdz · Mar 21, 2024 · f5642cd · f5642cd
1 parent 036fd0c
commit f5642cd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/amarillo/services/trips.py b/amarillo/services/trips.py
@@ -242,7 +242,7 @@ def enhance_carpool(self, carpool):
         if len(virtual_stops) > MAX_STOPS_PER_TRIP:
             # in case we found more than MAX_STOPS_PER_TRIP, we retain first and last 
             # half of MAX_STOPS_PER_TRIP
-            virtual_stops = virtual_stops.iloc[np.r_[0:int(MAX_STOPS_PER_TRIP/2), int(MAX_STOPS_PER_TRIP/2):]]
+            virtual_stops = virtual_stops.iloc[np.r_[0:int(MAX_STOPS_PER_TRIP/2), -int(MAX_STOPS_PER_TRIP/2):0]]
 
         trip_id = f"{carpool.agency}:{carpool.id}"
         stop_times = self._stops_and_stop_times(carpool.departureTime, trip_id, virtual_stops)