fix shuffle

3 years ago · 31ad4ba044
parent d60887fa6a
commit 31ad4ba044
1 changed files with 1 additions and 1 deletions
--- a/src/data/dataset.py
+++ b/src/data/dataset.py
@ -272,7 +272,7 @@ class TerrainDataset(Dataset):
        # * Randomize
        if self.randomize:
            np.random.seed(self.random_state)
-            blocks = np.random.shuffle(blocks)
+            np.random.shuffle(blocks)

        # * Remove blocks that contain nans
        mask = ~np.isnan(blocks).any(axis=1).any(axis=1)