From 45eed286ec595e283c24c713d3cffb411ea03ad6 Mon Sep 17 00:00:00 2001 From: rizoudal Date: Sun, 15 Sep 2024 15:36:04 +0200 Subject: [PATCH] refactor: comment out resize parameter to resolve ValueError in tests --- tests/test_losses.py | 2 +- tests/test_neuralnetwork.py | 2 +- tests/test_utils.py | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) 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) #-------------------------------------------------#