diff --git a/tests/test_losses.py b/tests/test_losses.py index c96b5e7b..60e508e8 100644 --- a/tests/test_losses.py +++ b/tests/test_losses.py @@ -57,7 +57,7 @@ def setUpClass(self): self.labels_ohe[i][class_index] = 1 # Create Data Generator self.datagen = DataGenerator(self.sampleList, self.tmp_data.name, - labels=self.labels_ohe, resize=(32, 32), + labels=self.labels_ohe, #resize=(32, 32), grayscale=False, batch_size=1) #-------------------------------------------------# diff --git a/tests/test_neuralnetwork.py b/tests/test_neuralnetwork.py index 1effd3e5..29ecc38a 100644 --- a/tests/test_neuralnetwork.py +++ b/tests/test_neuralnetwork.py @@ -60,7 +60,7 @@ def setUpClass(self): self.datagen = DataGenerator(self.sampleList_rgb, self.tmp_data.name, labels=self.labels_ohe, - resize=(32, 32), + #resize=(32, 32), shuffle=True, grayscale=False, batch_size=3) diff --git a/tests/test_utils.py b/tests/test_utils.py index 8cfb78be..b3705822 100644 --- a/tests/test_utils.py +++ b/tests/test_utils.py @@ -62,7 +62,7 @@ def setUpClass(self): self.datagen = DataGenerator(self.sampleList_rgb, self.tmp_data.name, labels=self.labels_ohe, - resize=(32, 32), + #resize=(32, 32), grayscale=False, batch_size=1) #-------------------------------------------------#