Set _artifact_uri when mlflow_run is not None. (microsoft#1367)

* Set _artifact_uri when mlflow_run is not None.

* Fix black.

qlib/workflow/recorder.py

@@ -279,6 +279,7 @@ def __init__(self, experiment_id, uri, name=None, mlflow_run=None):
                 if mlflow_run.info.end_time is not None
                 else None
             )
+            self._artifact_uri = mlflow_run.info.artifact_uri
         self.async_log = None
 
     def __repr__(self):

tests/test_workflow.py

@@ -0,0 +1,29 @@
+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT License.
+import unittest
+from pathlib import Path
+import shutil
+
+from qlib.workflow import R
+from qlib.tests import TestAutoData
+
+
+class WorkflowTest(TestAutoData):
+    TMP_PATH = Path("./.mlruns_tmp/")
+
+    def tearDown(self) -> None:
+        if self.TMP_PATH.exists():
+            shutil.rmtree(self.TMP_PATH)
+
+    def test_get_local_dir(self):
+        """ """
+        with R.start(uri=str(self.TMP_PATH)):
+            pass
+
+        with R.uri_context(uri=str(self.TMP_PATH)):
+            resume_recorder = R.get_recorder()
+            resume_recorder.get_local_dir()
+
+
+if __name__ == "__main__":
+    unittest.main()