Merge branch 'feat/dynamo_export_onnx' of https://github.com/GdoongMathew/pytorch-lightning into feat/dynamo_export_onnx

Author: Borda

src/lightning/pytorch/core/module.py

@@ -61,7 +61,7 @@
 from lightning.pytorch.trainer.connectors.logger_connector.result import _get_default_dtype
 from lightning.pytorch.utilities import GradClipAlgorithmType
 from lightning.pytorch.utilities.exceptions import MisconfigurationException
-from lightning.pytorch.utilities.imports import _TORCHMETRICS_GREATER_EQUAL_0_9_1, _TORCH_GREATER_EQUAL_2_6
+from lightning.pytorch.utilities.imports import _TORCH_GREATER_EQUAL_2_6, _TORCHMETRICS_GREATER_EQUAL_0_9_1
 from lightning.pytorch.utilities.model_helpers import _restricted_classmethod
 from lightning.pytorch.utilities.rank_zero import WarningCache, rank_zero_warn
 from lightning.pytorch.utilities.signature_utils import is_param_in_hook_signature

src/lightning/pytorch/utilities/imports.py

@@ -17,7 +17,7 @@
 import operator
 import sys
 
-from lightning_utilities.core.imports import RequirementCache, package_available, compare_version
+from lightning_utilities.core.imports import RequirementCache, compare_version, package_available
 
 from lightning.pytorch.utilities.rank_zero import rank_zero_warn
 

tests/tests_pytorch/models/test_onnx.py

@@ -25,10 +25,10 @@
 from lightning_utilities import compare_version
 
 import tests_pytorch.helpers.pipelines as tpipes
-from lightning.pytorch.utilities.imports import _TORCH_GREATER_EQUAL_2_6
 from lightning.pytorch import Trainer
 from lightning.pytorch.core.module import _ONNXSCRIPT_AVAILABLE
 from lightning.pytorch.demos.boring_classes import BoringModel
+from lightning.pytorch.utilities.imports import _TORCH_GREATER_EQUAL_2_6
 from tests_pytorch.helpers.runif import RunIf
 from tests_pytorch.utilities.test_model_summary import UnorderedModel