def on_run_end(self) -> Optional[_PREDICT_OUTPUT]:
        """Calls ``on_predict_epoch_end`` and ``on_predict_end`` hooks and returns results from all dataloaders."""
        results = self._on_predict_epoch_end()
        self._on_predict_end()
        self._on_predict_model_train()
        return results
Calls ``on_predict_epoch_end`` and ``on_predict_end`` hooks and returns results from all dataloaders.
src/lightning/pytorch/loops/prediction_loop.py