Typo fix: on_training_epoch_end -> on_train_epoch_end (#17110)

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>

Author: adamjstewart

docs/source-pytorch/common/lightning_module.rst

@@ -226,7 +226,7 @@ Train Epoch-level Operations
 ============================
 
 In the case that you need to make use of all the outputs from each :meth:`~lightning.pytorch.LightningModule.training_step`,
-override the :meth:`~lightning.pytorch.LightningModule.on_training_epoch_end` method.
+override the :meth:`~lightning.pytorch.LightningModule.on_train_epoch_end` method.
 
 .. code-block:: python
 

docs/source-pytorch/upgrade/sections/1_4_regular.rst

@@ -7,7 +7,7 @@
      - Ref
 
    * - relied on the ``outputs`` in your  ``LightningModule.on_train_epoch_end`` or ``Callback.on_train_epoch_end`` hooks
-     - rely on either ``on_training_epoch_end`` or set outputs as attributes in your ``LightningModule`` instances and access them from the hook
+     - rely on either ``on_train_epoch_end`` or set outputs as attributes in your ``LightningModule`` instances and access them from the hook
      - #7339
 
    * - accessed ``Trainer.truncated_bptt_steps``

docs/source-pytorch/upgrade/sections/1_9_advanced.rst

@@ -15,7 +15,7 @@
      - #16748
 
    * - implemented ``LightningModule.training_epoch_end`` hooks
-     - port your logic to  ``LightningModule.on_training_epoch_end`` hook
+     - port your logic to  ``LightningModule.on_train_epoch_end`` hook
      - #16520
 
    * - implemented ``LightningModule.validation_epoch_end`` hook

tests/tests_pytorch/trainer/logging_/test_eval_loop_logging.py

@@ -597,7 +597,7 @@ def training_step(self, batch, batch_idx):
             self.log("batch_idx", value, on_step=True, on_epoch=True, prog_bar=True)
             return out
 
-        def on_training_epoch_end(self):
+        def on_train_epoch_end(self):
             metrics = self.trainer.progress_bar_metrics
             v = 15 if self.current_epoch == 0 else 150
             assert metrics["batch_idx_epoch"] == (v / 5.0)