[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/CHANGELOG.md

@@ -11,7 +11,7 @@ The format is based on [Keep a Changelog](http://keepachangelog.com/en/1.0.0/).
 
 - Add enable_autolog_hparams argument to Trainer ([#20593](https://github.com/Lightning-AI/pytorch-lightning/pull/20593))
 
-- Add `toggled_optimizer(optimizer)` method to the LightningModule, which is a context manager version of `toggle_optimize` and `untoggle_optimizer` 
+- Add `toggled_optimizer(optimizer)` method to the LightningModule, which is a context manager version of `toggle_optimize` and `untoggle_optimizer`
 
 
 ### Changed

src/lightning/pytorch/core/module.py

@@ -1144,7 +1144,7 @@ def untoggle_optimizer(self, optimizer: Union[Optimizer, LightningOptimizer]) ->
     @contextmanager
     def toggled_optimizer(self, optimizer: Union[Optimizer, LightningOptimizer]) -> Generator:
         """Makes sure only the gradients of the current optimizer's parameters are calculated in the training step to
-        prevent dangling gradients in multiple-optimizer setup. Combines :meth:`toggle_optimizer` and 
+        prevent dangling gradients in multiple-optimizer setup. Combines :meth:`toggle_optimizer` and
         :meth:`untoggle_optimizer` into context manager.
 
         Args:
@@ -1159,6 +1159,7 @@ def training_step(...):
                     opt.zero_grad()
                     self.manual_backward(loss)
                     opt.step()
+
         """
         try:
             yield self.toggle_optimizer(optimizer)

tests/tests_pytorch/core/test_lightning_module.py

@@ -118,6 +118,7 @@ def test_1_optimizer_toggle_model():
     model.untoggle_optimizer(optimizer)
     assert not model._param_requires_grad_state
 
+
 def test_1_optimizer_toggle_model_context_manager():
     """Test toggle_model runs when only one optimizer is used."""
     model = BoringModel()