Merge pull request #18 from GreeleyGroup/fix/samplers

FIX: Fixed randomizer in RandomSampler
GreeleyGroup · Sep 23, 2023 · 2781f1a · 2781f1a
2 parents a12e497 + 64ea83c
commit 2781f1a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/samplers.py b/src/samplers.py
@@ -64,7 +64,7 @@ def create_samplers(self, sample_config):
         idx_array = np.arange(self.dataset_size)
 
         # Shuffle array
-        np.random.shuffle(idx_array)
+        randomizer.shuffle(idx_array)
 
         # Get indices
         train_size = int(np.ceil(sample_config["train"] * self.dataset_size))