[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

Author: pre-commit-ci[bot]

src/lightning/fabric/cli.py

@@ -29,8 +29,8 @@
 from lightning.fabric.utilities.consolidate_checkpoint import _process_cli_args
 from lightning.fabric.utilities.device_parser import _parse_gpu_ids
 from lightning.fabric.utilities.distributed import _suggested_max_num_threads
-from lightning.fabric.utilities.load import _load_distributed_checkpoint
 from lightning.fabric.utilities.imports import _lightning_xpu_available
+from lightning.fabric.utilities.load import _load_distributed_checkpoint
 
 _log = logging.getLogger(__name__)
 

src/lightning/fabric/utilities/imports.py

@@ -29,13 +29,16 @@
 _TORCH_GREATER_EQUAL_2_1 = compare_version("torch", operator.ge, "2.1.0", use_base_version=True)
 _TORCH_GREATER_EQUAL_2_2 = compare_version("torch", operator.ge, "2.2.0", use_base_version=True)
 _TORCH_GREATER_EQUAL_2_3 = compare_version("torch", operator.ge, "2.3.0", use_base_version=True)
-_TORCH_EQUAL_2_0 = compare_version("torch", operator.ge, "2.0.0", use_base_version=True) and not _TORCH_GREATER_EQUAL_2_1
+_TORCH_EQUAL_2_0 = (
+    compare_version("torch", operator.ge, "2.0.0", use_base_version=True) and not _TORCH_GREATER_EQUAL_2_1
+)
 
 _PYTHON_GREATER_EQUAL_3_8_0 = (sys.version_info.major, sys.version_info.minor) >= (3, 8)
 _PYTHON_GREATER_EQUAL_3_10_0 = (sys.version_info.major, sys.version_info.minor) >= (3, 10)
 
 _UTILITIES_GREATER_EQUAL_0_10 = compare_version("lightning_utilities", operator.ge, "0.10.0")
 
+
 @functools.lru_cache(maxsize=128)
 def _try_import_module(module_name: str) -> bool:
     try:

src/lightning/pytorch/trainer/connectors/accelerator_connector.py

@@ -62,8 +62,7 @@
 )
 from lightning.pytorch.strategies.ddp import _DDP_FORK_ALIASES
 from lightning.pytorch.utilities.exceptions import MisconfigurationException
-from lightning.pytorch.utilities.imports import _habana_available_and_importable
-from lightning.pytorch.utilities.imports import _lightning_xpu_available
+from lightning.pytorch.utilities.imports import _habana_available_and_importable, _lightning_xpu_available
 from lightning.pytorch.utilities.rank_zero import rank_zero_info, rank_zero_warn
 
 log = logging.getLogger(__name__)

src/lightning/pytorch/trainer/setup.py

@@ -28,8 +28,7 @@
     XLAProfiler,
 )
 from lightning.pytorch.utilities.exceptions import MisconfigurationException
-from lightning.pytorch.utilities.imports import _habana_available_and_importable
-from lightning.pytorch.utilities.imports import _lightning_xpu_available
+from lightning.pytorch.utilities.imports import _habana_available_and_importable, _lightning_xpu_available
 from lightning.pytorch.utilities.rank_zero import rank_zero_info, rank_zero_warn
 
 

src/lightning/pytorch/utilities/imports.py

@@ -49,6 +49,7 @@ def _habana_available_and_importable() -> bool:
     # also imports Lightning
     return bool(_LIGHTNING_HABANA_AVAILABLE) and _try_import_module("lightning_habana")
 
+
 _LIGHTNING_XPU_AVAILABLE = RequirementCache("lightning-xpu")