Skip to content

Commit

Permalink
Use the same DataFrame in the filter that is being filtered (#166)
Browse files Browse the repository at this point in the history
This is an completely untested proposed fix for #165
  • Loading branch information
bryanculbertson authored Jan 13, 2021
1 parent ef4a97f commit 31d6d26
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion peartree/parallel.py
Original file line number Diff line number Diff line change
Expand Up @@ -123,7 +123,7 @@ def generate_wait_times(
for direction in [0, 1]:
# Check if direction_id exists in source data
if 'direction_id' in trips_and_stop_times:
constraint_2 = (trips_and_stop_times.direction_id == direction)
constraint_2 = (stop_times_by_stop.direction_id == direction)
direction_subset = stop_times_by_stop[constraint_2]
else:
direction_subset = stop_times_by_stop.copy()
Expand Down

0 comments on commit 31d6d26

Please sign in to comment.