fix pytest error

microsoft · Dec 13, 2024 · 202fc96 · 202fc96
1 parent 0dc0f38
commit 202fc96
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/qlib/workflow/recorder.py b/qlib/workflow/recorder.py
@@ -9,6 +9,7 @@
 import pickle
 import tempfile
 import subprocess
+import platform
 from pathlib import Path
 from datetime import datetime
 
@@ -316,7 +317,10 @@ def get_local_dir(self):
         This function will return the directory path of this recorder.
         """
         if self.artifact_uri is not None:
-            local_dir_path = Path(self.artifact_uri.lstrip("file:").lstrip("/")).parent
+            if platform.system() == "Linux":
+                local_dir_path = Path(self.artifact_uri.lstrip("file:")).parent
+            else:
+                local_dir_path = Path(self.artifact_uri.lstrip("file:").lstrip("/")).parent
             local_dir_path = str(local_dir_path.resolve())
             if os.path.isdir(local_dir_path):
                 return local_dir_path