fix: Incorrect URI Prefix Stripping in MLFlowLogger (#20365)

Lightning-AI · Nov 13, 2024 · cae3335 · cae3335
1 parent 875cfbd
commit cae3335
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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