Fixed code formatting.

bakert1 · Feb 26, 2024 · 7574aa0 · 7574aa0
1 parent 420a2d1
commit 7574aa0
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 7 deletions.
diff --git a/monai/transforms/intensity/array.py b/monai/transforms/intensity/array.py
@@ -98,11 +98,12 @@ class RandGaussianNoise(RandomizableTransform):
     backend = [TransformBackends.TORCH, TransformBackends.NUMPY]
 
     def __init__(
-            self, prob: float = 0.1,
-            mean: float = 0.0,
-            std: float = 0.1,
-            dtype: DtypeLike = np.float32,
-            sample_std: bool = True,
+        self,
+        prob: float = 0.1,
+        mean: float = 0.0,
+        std: float = 0.1,
+        dtype: DtypeLike = np.float32,
+        sample_std: bool = True,
     ) -> None:
         RandomizableTransform.__init__(self, prob)
         self.mean = mean

diff --git a/monai/transforms/intensity/dictionary.py b/monai/transforms/intensity/dictionary.py
@@ -196,7 +196,7 @@ def __init__(
         std: float = 0.1,
         dtype: DtypeLike = np.float32,
         allow_missing_keys: bool = False,
-        sample_std: bool = True
+        sample_std: bool = True,
     ) -> None:
         MapTransform.__init__(self, keys, allow_missing_keys)
         RandomizableTransform.__init__(self, prob)

diff --git a/tests/test_rand_gaussian_noised.py b/tests/test_rand_gaussian_noised.py
@@ -33,7 +33,9 @@ class TestRandGaussianNoised(NumpyImageTestCase2D):
 
     @parameterized.expand(TESTS)
     def test_correct_results(self, _, im_type, keys, mean, std, sample_std):
-        gaussian_fn = RandGaussianNoised(keys=keys, prob=1.0, mean=mean, std=std, dtype=np.float64, sample_std=sample_std)
+        gaussian_fn = RandGaussianNoised(
+            keys=keys, prob=1.0, mean=mean, std=std, dtype=np.float64, sample_std=sample_std
+        )
         gaussian_fn.set_random_state(seed)
         im = im_type(self.imt)
         noised = gaussian_fn({k: im for k in keys})