From 64ea83c511060f74eac5e1de238fe791ec5f61c4 Mon Sep 17 00:00:00 2001 From: Gaurav S Deshmukh Date: Fri, 22 Sep 2023 23:29:53 -0400 Subject: [PATCH] Added fix --- src/samplers.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/samplers.py b/src/samplers.py index 7723ae1..d6d79ef 100644 --- 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))