update tests

nitrain · May 6, 2024 · 1fefb9e · 1fefb9e
1 parent 383ea3f
commit 1fefb9e
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 12 deletions.
diff --git a/tests/run_tests.sh b/tests/run_tests.sh
@@ -37,7 +37,7 @@ $PYCMD test_readers.py $@
 
 echo "Testing samplers"
 $PYCMD test_samplers.py $@
-$PYCMD test_samplers_slice.py $@
+$PYCMD test_samplers_slices.py $@
 
 echo "Testing trainers"
 $PYCMD test_trainers.py $@

diff --git a/tests/test_workflows.py b/tests/test_workflows.py
@@ -80,17 +80,17 @@ def test_multiclass_segmentation(self):
 
        # data_train, data_test = dataset.split(0.8)
 
-        loader = nt.Loader(dataset,
-                           images_per_batch=4)
-
-        xb, yb = next(iter(loader))
-
-        arch_fn = nt.fetch_architecture('unet', dim=2)
-        model = arch_fn(x.shape[:-1]+(1,),
-                        number_of_outputs=2,
-                        number_of_layers=4,
-                        number_of_filters_at_base_layer=16,
-                        mode='classification')
+        #loader = nt.Loader(dataset,
+        #                   images_per_batch=4)
+        #
+        #xb, yb = next(iter(loader))
+#
+        #arch_fn = nt.fetch_architecture('unet', dim=2)
+        #model = arch_fn(x.shape[:-1]+(1,),
+        #                number_of_outputs=2,
+        #                number_of_layers=4,
+        #                number_of_filters_at_base_layer=16,
+        #                mode='classification')
 #
         ## train
         #trainer = nt.Trainer(model, task='segmentation')