fix num_aug=16

Author: donglaiw

connectomics/data/augmentation/test_augmentor.py

@@ -82,16 +82,15 @@ def _tta_3d(self, model, data):
         out = None
         cc = 0
 
-        if self.num_aug == None:
-            opts = itertools.product(
-                (False, ), (False, ), (False, ), (False, ))
-        elif self.num_aug == 4:
+        opts = itertools.product(
+            (False, ), (False, ), (False, ), (False, ))
+        if self.num_aug == 4:
             opts = itertools.product(
                 (False, True), (False, True), (False, ), (False, ))
         elif self.num_aug == 8:
             opts = itertools.product(
                 (False, True), (False, True), (False, ), (False, True))
-        else:
+        elif self.num_aug == 16:
             opts = itertools.product(
                 (False, True), (False, True), (False, True), (False, True))
 
@@ -210,7 +209,7 @@ def update_name(self, name):
         r"""Update the name of the output file to indicate applied test-time augmentations.
         """
         extension = "_"
-        if self.num_aug is None:
+        if self.num_aug is None
             return name
         elif self.num_aug == 4:
             extension += "xy"