fix: wrong index when comparing;

WenjieDu · Jun 17, 2024 · 1f8d102 · 1f8d102
1 parent d041eb2
commit 1f8d102
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pypots/data/utils.py b/pypots/data/utils.py
@@ -197,8 +197,8 @@ def sliding_window(time_series, window_len, sliding_len=None):
  start_indices = np.asarray(range(total_len // sliding_len)) * sliding_len
 
  # remove the last one if left length is not enough
- if total_len - start_indices[-1] * sliding_len < window_len:
- to_drop = math.ceil(window_len / sliding_len) - 1
+ if total_len - start_indices[-1] < window_len:
+ to_drop = math.ceil(window_len / sliding_len)
  left_len = total_len - start_indices[-1]
  start_indices = start_indices[:-to_drop]
  logger.warning(