From 31ad4ba0441572349a2c2386c8c3a988d6f04a44 Mon Sep 17 00:00:00 2001 From: Deniz Ugur Date: Sun, 16 May 2021 00:59:07 +0300 Subject: [PATCH] fix shuffle --- src/data/dataset.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/data/dataset.py b/src/data/dataset.py index e2914f7..fda42f0 100644 --- 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)