Skip to content

Commit d519ba8

Browse files
authored
Merge pull request #108 from aanil/master
Replace numpy.split with iloc to get rid of FutureWarning on using Series.swapaxes
2 parents 63a1516 + 929086b commit d519ba8

File tree

1 file changed

+14
-1
lines changed

1 file changed

+14
-1
lines changed

sensorpush_to_statusdb.py

Lines changed: 14 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -177,14 +177,27 @@ def summarize_intervals(self, sample_series, limit_type):
177177

178178
interval_points = []
179179
extended_intervals = []
180-
for interval in np.split(sample_series, gap_positions):
180+
# Initialize the starting index for slicing
181+
start_idx = 0
182+
for end_idx in gap_positions:
183+
interval = sample_series.iloc[start_idx:end_idx]
181184
lower = interval.index[0]
182185
upper = interval.index[-1]
183186
interval_points.append((lower, upper))
184187
# Extended interval with 1 hour in each direction
185188
extend_lower = lower - np.timedelta64(1, "h")
186189
extend_upper = upper + np.timedelta64(1, "h")
187190
extended_intervals.append((extend_lower, extend_upper))
191+
start_idx = end_idx
192+
193+
# Capture the last interval
194+
interval = sample_series.iloc[start_idx:]
195+
lower = interval.index[0]
196+
upper = interval.index[-1]
197+
interval_points.append((lower, upper))
198+
extend_lower = lower - np.timedelta64(1, "h")
199+
extend_upper = upper + np.timedelta64(1, "h")
200+
extended_intervals.append((extend_lower, extend_upper))
188201

189202
return interval_points, extended_intervals
190203

0 commit comments

Comments
 (0)