[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/loops/fit_loop.py

@@ -354,16 +354,20 @@ def reset(self) -> None:
         # TODO LUCA: refactor restarting for fit_loop
         restarting_mid_epoch = self.restarting_mid_epoch
 
-        if (self.epoch_loop.restarting_on_train_batch_end
+        if (
+            self.epoch_loop.restarting_on_train_batch_end
             and self.restarting_mid_epoch
-            and self.epoch_loop.batch_progress.is_last_batch):
+            and self.epoch_loop.batch_progress.is_last_batch
+        ):
             self.epoch_progress.increment_processed()
             self.epoch_progress.increment_completed()
 
-        if (self.epoch_loop.restarting_on_train_batch_end
+        if (
+            self.epoch_loop.restarting_on_train_batch_end
             and self.epoch_loop.batch_progress.is_last_batch
             and not restarting_mid_epoch
-            and not self.epoch_loop.val_loop.batch_progress.is_last_batch):
+            and not self.epoch_loop.val_loop.batch_progress.is_last_batch
+        ):
             self.epoch_progress.increment_completed()
 
     def on_run_start(self) -> None:

src/lightning/pytorch/loops/training_epoch_loop.py

@@ -182,7 +182,7 @@ def reset(self) -> None:
             # happens on_validation_end
             # we could set it in the checkpoint but we prefer to keep checkpoints backward compatible
             if self.restarting_on_train_batch_end or not self.restarting_on_last:
-            # if not self.restarting_on_train_batch_end and not self.restarting_on_last:
+                # if not self.restarting_on_train_batch_end and not self.restarting_on_last:
                 self._batches_that_stepped += 1
 
         if self.restarting_on_train_batch_end: