fixed code format checks

juampatronics · Mar 24, 2024 · c85976f · c85976f
1 parent c04c8c8
commit c85976f
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/monai/transforms/regularization/array.py b/monai/transforms/regularization/array.py
@@ -16,12 +16,12 @@
 
 import torch
 
-from monai.transforms import Randomizable, Transform
+from ..transform import RandomizableTransform
 
 __all__ = ["MixUp", "CutMix", "CutOut"]
 
 
-class Mixer(Transform, Randomizable):
+class Mixer(RandomizableTransform):
     def __init__(self, batch_size: int, alpha: float = 1.0) -> None:
         super().__init__()
         if alpha <= 0:

diff --git a/monai/transforms/regularization/dictionary.py b/monai/transforms/regularization/dictionary.py
@@ -12,9 +12,9 @@
 from __future__ import annotations
 
 from monai.config import KeysCollection
-from monai.transforms import MapTransform
 from monai.utils.misc import ensure_tuple
 
+from ..transform import MapTransform
 from .array import CutMix, CutOut, MixUp
 
 __all__ = ["MixUpd", "MixUpD", "MixUpDict", "CutMixd", "CutMixD", "CutMixDict", "CutOutd", "CutOutD", "CutOutDict"]