From 7574aa0c3e331c049a7945db1225bf146076c761 Mon Sep 17 00:00:00 2001 From: Timothy Baker Date: Sun, 25 Feb 2024 21:50:59 -0500 Subject: [PATCH] Fixed code formatting. --- monai/transforms/intensity/array.py | 11 ++++++----- monai/transforms/intensity/dictionary.py | 2 +- tests/test_rand_gaussian_noised.py | 4 +++- 3 files changed, 10 insertions(+), 7 deletions(-) diff --git a/monai/transforms/intensity/array.py b/monai/transforms/intensity/array.py index b6b4d45398..a2f63a7482 100644 --- 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 index 73e5ece3c7..7e93464e64 100644 --- 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 index 6b43274d09..e3df196be2 100644 --- 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})