specify boolean series type to avoid future warning

jarq6c · Feb 9, 2024 · 200bc03 · 200bc03
1 parent 3f39564
commit 200bc03
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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]