Merge pull request #46 from WenjieDu/dev

Avoid empty step_idx list when n_steps==seq_len
WenjieDu · Sep 6, 2024 · eeb0e64 · eeb0e64
2 parents 533b328 + eac2b26
commit eeb0e64
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(