fixed nbb boostrapping

cerlymarco · Nov 23, 2023 · 6eede02 · 6eede02
1 parent 8703fb9
commit 6eede02
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -3,7 +3,7 @@
 
 HERE = pathlib.Path(__file__).parent
 
-VERSION = '1.0.4'
+VERSION = '1.0.5'
 PACKAGE_NAME = 'tsmoothie'
 AUTHOR = 'Marco Cerliani'
 AUTHOR_EMAIL = 'cerlymarco@gmail.com'

diff --git a/tsmoothie/utils_func.py b/tsmoothie/utils_func.py
@@ -510,7 +510,8 @@ def _id_nb_bootstrap(n_obs, block_length):
         np.arange(0, n_obs, block_length)
     ).reshape(-1, 1)
 
-    _id = (blocks + nexts).ravel()[:n_obs]
+    _id = (blocks + nexts).ravel()
+    _id = _id[_id < n_obs]
 
     return _id