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

MrWhatZitToYaa · Nov 20, 2024 · 31bd74a · 31bd74a
1 parent bec81bc
commit 31bd74a
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