fix num_aug in the constructor

Author: donglaiw

connectomics/data/augmentation/test_augmentor.py

@@ -54,8 +54,10 @@ def __init__(self,
         self.do_2d = do_2d
         self.scale_factors = scale_factors
         self.inference_act = inference_act
+        if num_aug == 0:
+            num_aug = None:
 
-        if num_aug is not None and num_aug != 0:
+        if num_aug is not None:
             assert num_aug in [4, 8, 16], \
                 "TestAugmentor.num_aug should be either 4, 8 or 16!"
             if self.do_2d: # max num_aug for 2d images
@@ -208,7 +210,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 or self.num_aug == 0:
+        if self.num_aug is None:
             return name
         elif self.num_aug == 4:
             extension += "xy"