Minor cleanup.

Signed-off-by: Michal Zientkiewicz <michalz@nvidia.com>
NVIDIA · Nov 4, 2024 · 628e146 · 628e146
1 parent 352d3da
commit 628e146
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 7 deletions.
diff --git a/dali/test/python/checkpointing/test_dali_stateless_operators.py b/dali/test/python/checkpointing/test_dali_stateless_operators.py
@@ -27,8 +27,7 @@
     restrict_platform,
 )
 from nose2.tools import params, cartesian_params
-from nose_utils import assert_raises, SkipTest
-from nose_utils import attr
+from nose_utils import assert_raises, SkipTest, attr
 
 # Test configuration
 batch_size = 8

diff --git a/dali/test/python/decoder/test_image.py b/dali/test/python/decoder/test_image.py
@@ -23,7 +23,7 @@
 from nvidia.dali import pipeline_def
 
 from nose2.tools import params
-from nose_utils import assert_raises
+from nose_utils import assert_raises, SkipTest
 from test_utils import check_output_pattern
 from test_utils import compare_pipelines
 from test_utils import get_dali_extra_path
@@ -279,9 +279,8 @@ def check_fancy_upsampling_body(batch_size, img_type, device):
 @params(1, 8)
 def test_fancy_upsampling(batch_size):
     if nvidia.dali.backend.GetNvjpegVersion() < 12001:
-        from nose_utils import SkipTest
-
         raise SkipTest("nvJPEG doesn't support fancy upsampling in this version")
+
     data_path = os.path.join(test_data_root, good_path, "jpeg")
     compare_pipelines(
         decoder_pipe(

diff --git a/dali/test/python/decoder/test_imgcodec.py b/dali/test/python/decoder/test_imgcodec.py
@@ -280,9 +280,8 @@ def check_fancy_upsampling_body(batch_size, img_type, device):
 @params(1, 8)
 def test_fancy_upsampling(batch_size):
     if nvidia.dali.backend.GetNvjpegVersion() < 12001:
-        from nose_utils import SkipTest
-
         raise SkipTest("nvJPEG doesn't support fancy upsampling in this version")
+
     data_path = os.path.join(test_data_root, good_path, "jpeg")
     compare_pipelines(
         decoder_pipe(