include refac

src/data_handling/bootstraps.py

@@ -176,7 +176,7 @@ class BootStraps(RunEnvironment):
         return False, max_nboot
 
     @staticmethod
-    def shuffle_single_variable(data: da.array, chunks) -> np.ndarray:
+    def shuffle_single_variable(data: da.array, chunks) -> da.core.Array:
         size = data.shape
         return da.random.choice(data.reshape(-1,), size=size, chunks=chunks)
 

test/test_data_handling/test_bootstraps.py

@@ -54,7 +54,7 @@ class TestBootstraps:
 
     def test_shuffle_single_variale(self, boot_no_init):
         data = np.array([[1, 2, 3], [1, 2, 3], [1, 2, 3], [1, 2, 3]])
-        res = boot_no_init.shuffle_single_variable(data)
+        res = boot_no_init.shuffle_single_variable(data, chunks=(2, 3)).compute()
         assert res.shape == data.shape
         assert res.max() == data.max()
         assert res.min() == data.min()