fix: avoid empty step_idx list when n_steps==seq_len;

WenjieDu · Sep 3, 2024 · eac2b26 · eac2b26
1 parent 16aa3e5
commit eac2b26
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pygrinder/sequential_missing/seq_missing.py b/pygrinder/sequential_missing/seq_missing.py
@@ -166,7 +166,7 @@ def seq_missing(
             n_steps - max(step_idx) >= seq_len
         ), f"n_steps - max(step_idx) must be >= seq_len, but got {n_steps - max(step_idx)}"
     else:
-        step_idx = list(range(n_steps - seq_len))
+        step_idx = list(range(n_steps - seq_len + 1))
 
     if isinstance(X, np.ndarray):
         corrupted_X = _seq_missing_numpy(