[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/pytorch/trainer/trainer.py

@@ -28,8 +28,10 @@
 from typing import Any, Dict, Generator, Iterable, List, Optional, Union
 from weakref import proxy
 
-import lightning.pytorch as pl
 import torch
+from torch.optim import Optimizer
+
+import lightning.pytorch as pl
 from lightning.fabric.utilities.apply_func import convert_tensors_to_scalars
 from lightning.fabric.utilities.cloud_io import _is_local_file_protocol
 from lightning.fabric.utilities.types import _PATH
@@ -77,7 +79,6 @@
     LRSchedulerConfig,
 )
 from lightning.pytorch.utilities.warnings import PossibleUserWarning
-from torch.optim import Optimizer
 
 log = logging.getLogger(__name__)
 

tests/tests_pytorch/callbacks/test_finetuning_callback.py

@@ -19,11 +19,12 @@
 from lightning.pytorch import LightningModule, Trainer, seed_everything
 from lightning.pytorch.callbacks import BackboneFinetuning, BaseFinetuning, ModelCheckpoint
 from lightning.pytorch.demos.boring_classes import BoringModel, RandomDataset
-from tests_pytorch.helpers.runif import RunIf
 from torch import nn
 from torch.optim import SGD, Optimizer
 from torch.utils.data import DataLoader
 
+from tests_pytorch.helpers.runif import RunIf
+
 
 class TestBackboneFinetuningCallback(BackboneFinetuning):
     def on_train_epoch_start(self, trainer, pl_module):
@@ -282,12 +283,10 @@ def test_complex_nested_model():
     directly themselves rather than exclusively their submodules containing parameters."""
 
     model = nn.Sequential(
-        OrderedDict(
-            [
-                ("encoder", nn.Sequential(ConvBlockParam(3, 64), ConvBlock(64, 128))),
-                ("decoder", ConvBlock(128, 10)),
-            ]
-        )
+        OrderedDict([
+            ("encoder", nn.Sequential(ConvBlockParam(3, 64), ConvBlock(64, 128))),
+            ("decoder", ConvBlock(128, 10)),
+        ])
     )
 
     # There are 10 leaf modules or parent modules w/ parameters in the test model
@@ -437,8 +436,8 @@ def test_unsupported_strategies(tmp_path):
 
 
 def test_finetuning_with_configure_model(tmp_path):
-    """Test that BaseFinetuning works correctly with configure_model by ensuring freeze_before_training
-    is called after configure_model but before training starts."""
+    """Test that BaseFinetuning works correctly with configure_model by ensuring freeze_before_training is called after
+    configure_model but before training starts."""
 
     class TrackingFinetuningCallback(BaseFinetuning):
         def __init__(self):