refactor: comment out resize parameter to resolve ValueError in tests

frankkramer-lab · Sep 15, 2024 · 45eed28 · 45eed28
1 parent afa71bb
commit 45eed28
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git 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
@@ -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
@@ -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)
 
     #-------------------------------------------------#