diff --git a/python/events/src/hydrotools/events/event_detection/decomposition.py b/python/events/src/hydrotools/events/event_detection/decomposition.py index 1c9c3f30..8fbc3e99 100644 --- a/python/events/src/hydrotools/events/event_detection/decomposition.py +++ b/python/events/src/hydrotools/events/event_detection/decomposition.py @@ -153,12 +153,12 @@ def event_boundaries(event_points: pd.Series): """ # Identify event starts - forward_shift = event_points.shift(1).fillna(False) + forward_shift = event_points.shift(1).astype(bool).fillna(False) starts = (event_points & ~forward_shift) starts = starts[starts] # Identify event ends - backward_shift = event_points.shift(-1).fillna(False) + backward_shift = event_points.shift(-1).astype(bool).fillna(False) ends = (event_points & ~backward_shift) ends = ends[ends]