fix format

Signed-off-by: YunLiu <[email protected]>

Author: KumoLiu

monai/utils/__init__.py

@@ -107,6 +107,7 @@
     InvalidPyTorchVersionError,
     OptionalImportError,
     allow_missing_reference,
+    compute_capabilities_after,
     damerau_levenshtein_distance,
     exact_version,
     get_full_type_name,
@@ -123,7 +124,6 @@
     run_eval,
     version_geq,
     version_leq,
-    compute_capabilities_after,
 )
 from .nvtx import Range
 from .ordering import Ordering

monai/utils/module.py

@@ -26,8 +26,8 @@
 from re import match
 from types import FunctionType, ModuleType
 from typing import Any, cast
-import pynvml
 
+import pynvml
 import torch
 
 # bundle config system flags
@@ -636,6 +636,7 @@ def pytorch_after(major: int, minor: int, patch: int = 0, current_ver_string: st
         return False
     return True
 
+
 @functools.lru_cache(None)
 def compute_capabilities_after(major: int, minor: int = 0, current_ver_string: str | None = None) -> bool:
     """

tests/test_bundle_trt_export.py

@@ -22,7 +22,13 @@
 from monai.data import load_net_with_metadata
 from monai.networks import save_state
 from monai.utils import optional_import
-from tests.utils import command_line_tests, skip_if_no_cuda, skip_if_quick, skip_if_windows, SkipIfBeforeComputeCapabilityVersion
+from tests.utils import (
+    SkipIfBeforeComputeCapabilityVersion,
+    command_line_tests,
+    skip_if_no_cuda,
+    skip_if_quick,
+    skip_if_windows,
+)
 
 _, has_torchtrt = optional_import(
     "torch_tensorrt",

tests/test_convert_to_trt.py

@@ -20,7 +20,7 @@
 from monai.networks import convert_to_trt
 from monai.networks.nets import UNet
 from monai.utils import optional_import
-from tests.utils import skip_if_no_cuda, skip_if_quick, skip_if_windows, SkipIfBeforeComputeCapabilityVersion
+from tests.utils import SkipIfBeforeComputeCapabilityVersion, skip_if_no_cuda, skip_if_quick, skip_if_windows
 
 _, has_torchtrt = optional_import(
     "torch_tensorrt",

tests/test_trt_compile.py

@@ -21,7 +21,13 @@
 from monai.networks import trt_compile
 from monai.networks.nets import UNet, cell_sam_wrapper, vista3d132
 from monai.utils import min_version, optional_import
-from tests.utils import SkipIfAtLeastPyTorchVersion, skip_if_no_cuda, skip_if_quick, skip_if_windows, SkipIfBeforeComputeCapabilityVersion
+from tests.utils import (
+    SkipIfAtLeastPyTorchVersion,
+    SkipIfBeforeComputeCapabilityVersion,
+    skip_if_no_cuda,
+    skip_if_quick,
+    skip_if_windows,
+)
 
 trt, trt_imported = optional_import("tensorrt", "10.1.0", min_version)
 polygraphy, polygraphy_imported = optional_import("polygraphy")

tests/test_version_after.py

@@ -15,7 +15,7 @@
 
 from parameterized import parameterized
 
-from monai.utils import pytorch_after, compute_capabilities_after
+from monai.utils import compute_capabilities_after, pytorch_after
 
 TEST_CASES_PT = (
     (1, 5, 9, "1.6.0"),

tests/utils.py

@@ -47,7 +47,7 @@
 from monai.networks import convert_to_onnx, convert_to_torchscript
 from monai.utils import optional_import
 from monai.utils.misc import MONAIEnvVars
-from monai.utils.module import pytorch_after, compute_capabilities_after
+from monai.utils.module import compute_capabilities_after, pytorch_after
 from monai.utils.tf32 import detect_default_tf32
 from monai.utils.type_conversion import convert_data_type