flake8 warnings

juampatronics · Mar 25, 2024 · f2fe14e · f2fe14e
1 parent 1b97e69
commit f2fe14e
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 6 deletions.
diff --git a/monai/transforms/regularization/dictionary.py b/monai/transforms/regularization/dictionary.py
@@ -28,9 +28,7 @@ class MixUpd(MapTransform):
     for consistency, i.e. images and labels must be applied the same augmenation.
     """
 
-    def __init__(
-        self, keys: KeysCollection, batch_size: int, alpha: float = 1.0, allow_missing_keys: bool = False
-    ) -> None:
+    def __init__(self, keys: KeysCollection, batch_size: int, alpha: float = 1.0, allow_missing_keys: bool = False) -> None:
         super().__init__(keys, allow_missing_keys)
         self.mixup = MixUp(batch_size, alpha)
 

diff --git a/tests/test_regularization.py b/tests/test_regularization.py
@@ -26,7 +26,7 @@ def test_mixup(self):
             mixup = MixUp(6, 1.0)
             output = mixup(sample)
             self.assertEqual(output.shape, sample.shape)
-            self.assertTrue(any([not torch.allclose(sample, mixup(sample)) for _ in range(10)]))
+            self.assertTrue(any(not torch.allclose(sample, mixup(sample)) for _ in range(10)))
 
         with self.assertRaises(ValueError):
             MixUp(6, -0.5)
@@ -59,7 +59,7 @@ def test_cutmix(self):
             cutmix = CutMix(6, 1.0)
             output = cutmix(sample)
             self.assertEqual(output.shape, sample.shape)
-            self.assertTrue(any([not torch.allclose(sample, cutmix(sample)) for _ in range(10)]))
+            self.assertTrue(any(not torch.allclose(sample, cutmix(sample)) for _ in range(10)))
 
     def test_cutmixd(self):
         for dims in [2, 3]:
@@ -83,7 +83,7 @@ def test_cutout(self):
             cutout = CutOut(6, 1.0)
             output = cutout(sample)
             self.assertEqual(output.shape, sample.shape)
-            self.assertTrue(any([not torch.allclose(sample, cutout(sample)) for _ in range(10)]))
+            self.assertTrue(any(not torch.allclose(sample, cutout(sample)) for _ in range(10)))
 
 
 if __name__ == "__main__":