diff --git a/src/lightning/pytorch/loggers/mlflow.py b/src/lightning/pytorch/loggers/mlflow.py index 1b15014cd0528..ec990b634a6c4 100644 --- a/src/lightning/pytorch/loggers/mlflow.py +++ b/src/lightning/pytorch/loggers/mlflow.py @@ -299,7 +299,7 @@ def save_dir(self) -> Optional[str]: """ if self._tracking_uri.startswith(LOCAL_FILE_URI_PREFIX): - return self._tracking_uri.lstrip(LOCAL_FILE_URI_PREFIX) + return self._tracking_uri[len(LOCAL_FILE_URI_PREFIX) :] return None @property