diff --git a/src/lightning/pytorch/trainer/trainer.py b/src/lightning/pytorch/trainer/trainer.py index f0aba4d5ed..d0988d9d38 100644 --- a/src/lightning/pytorch/trainer/trainer.py +++ b/src/lightning/pytorch/trainer/trainer.py @@ -506,7 +506,7 @@ class Trainer: train_dataloaders: Optional[Union[TRAIN_DATALOADERS, LightningDataModule]] = None, val_dataloaders: Optional[EVAL_DATALOADERS] = None, datamodule: Optional[LightningDataModule] = None, - ckpt_path: Optional[str] = None, + ckpt_path: Optional[_PATH] = None, ) -> None: r"""Runs the full optimization routine. @@ -550,7 +550,7 @@ class Trainer: train_dataloaders: Optional[Union[TRAIN_DATALOADERS, LightningDataModule]] = None, val_dataloaders: Optional[EVAL_DATALOADERS] = None, datamodule: Optional[LightningDataModule] = None, - ckpt_path: Optional[str] = None, + ckpt_path: Optional[_PATH] = None, ) -> None: log.debug(f"{self.__class__.__name__}: trainer fit stage") @@ -586,7 +586,7 @@ class Trainer: self, model: Optional["pl.LightningModule"] = None, dataloaders: Optional[Union[EVAL_DATALOADERS, LightningDataModule]] = None, - ckpt_path: Optional[str] = None, + ckpt_path: Optional[_PATH] = None, verbose: bool = True, datamodule: Optional[LightningDataModule] = None, ) -> _EVALUATE_OUTPUT: @@ -649,7 +649,7 @@ class Trainer: self, model: Optional["pl.LightningModule"] = None, dataloaders: Optional[Union[EVAL_DATALOADERS, LightningDataModule]] = None, - ckpt_path: Optional[str] = None, + ckpt_path: Optional[_PATH] = None, verbose: bool = True, datamodule: Optional[LightningDataModule] = None, ) -> Optional[Union[_PREDICT_OUTPUT, _EVALUATE_OUTPUT]]: @@ -694,7 +694,7 @@ class Trainer: self, model: Optional["pl.LightningModule"] = None, dataloaders: Optional[Union[EVAL_DATALOADERS, LightningDataModule]] = None, - ckpt_path: Optional[str] = None, + ckpt_path: Optional[_PATH] = None, verbose: bool = True, datamodule: Optional[LightningDataModule] = None, ) -> _EVALUATE_OUTPUT: @@ -758,7 +758,7 @@ class Trainer: self, model: Optional["pl.LightningModule"] = None, dataloaders: Optional[Union[EVAL_DATALOADERS, LightningDataModule]] = None, - ckpt_path: Optional[str] = None, + ckpt_path: Optional[_PATH] = None, verbose: bool = True, datamodule: Optional[LightningDataModule] = None, ) -> Optional[Union[_PREDICT_OUTPUT, _EVALUATE_OUTPUT]]: @@ -805,7 +805,7 @@ class Trainer: dataloaders: Optional[Union[EVAL_DATALOADERS, LightningDataModule]] = None, datamodule: Optional[LightningDataModule] = None, return_predictions: Optional[bool] = None, - ckpt_path: Optional[str] = None, + ckpt_path: Optional[_PATH] = None, ) -> Optional[_PREDICT_OUTPUT]: r"""Run inference on your data. This will call the model forward function to compute predictions. Useful to perform distributed and batched predictions. Logging is disabled in the predict hooks. @@ -870,7 +870,7 @@ class Trainer: dataloaders: Optional[Union[EVAL_DATALOADERS, LightningDataModule]] = None, datamodule: Optional[LightningDataModule] = None, return_predictions: Optional[bool] = None, - ckpt_path: Optional[str] = None, + ckpt_path: Optional[_PATH] = None, ) -> Optional[_PREDICT_OUTPUT]: # -------------------- # SETUP HOOK