From bfc68a494ace8a367a61515d37e2d1c206032766 Mon Sep 17 00:00:00 2001 From: Rafael Dias Date: Thu, 21 Nov 2024 17:02:17 +0000 Subject: [PATCH] Style fix --- tests/test_bundle_ckpt_export.py | 2 +- tests/test_clip_intensity_percentilesd.py | 3 +-- 2 files changed, 2 insertions(+), 3 deletions(-) diff --git a/tests/test_bundle_ckpt_export.py b/tests/test_bundle_ckpt_export.py index 2dfabc0ec6..237e0e355b 100644 --- a/tests/test_bundle_ckpt_export.py +++ b/tests/test_bundle_ckpt_export.py @@ -96,7 +96,7 @@ def test_ckpt_export(self, key_in_ckpt, use_trace): self.assertTrue(os.path.exists(self.ts_file)) _, metadata, extra_files = load_net_with_metadata( - self.ts_file, more_extra_files=["inference.json", "def_args.json"] + self.ts_file, more_extra_files=["inference.json", "def_args.json"] ) self.assertIn("schema", metadata) self.assertIn("meta_file", json.loads(extra_files["def_args.json"])) diff --git a/tests/test_clip_intensity_percentilesd.py b/tests/test_clip_intensity_percentilesd.py index 853d7becf9..5e0eec4ced 100644 --- a/tests/test_clip_intensity_percentilesd.py +++ b/tests/test_clip_intensity_percentilesd.py @@ -18,9 +18,8 @@ from monai.transforms import ClipIntensityPercentilesd from monai.transforms.utils_pytorch_numpy_unification import clip, percentile from monai.utils.type_conversion import convert_to_tensor -from tests.utils import TEST_NDARRAYS, NumpyImageTestCase2D, NumpyImageTestCase3D, assert_allclose - from tests.test_clip_intensity_percentiles import test_hard_clip_func, test_soft_clip_func +from tests.utils import TEST_NDARRAYS, NumpyImageTestCase2D, NumpyImageTestCase3D, assert_allclose class TestClipIntensityPercentilesd2D(NumpyImageTestCase2D):