fix tmp_dir always being created

mlflow · Dec 30, 2021 · 82b1a1d · 82b1a1d
1 parent 62bd02b
commit 82b1a1d
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/mlflow/store/artifact/hdfs_artifact_repo.py b/mlflow/store/artifact/hdfs_artifact_repo.py
@@ -125,7 +125,10 @@ def download_artifacts(self, artifact_path, dst_path=None):
         """
 
         hdfs_base_path = _resolve_base_path(self.path, artifact_path)
-        local_dir = _tmp_dir(dst_path)
+        if dst_path and os.path.exists(dst_path):
+            local_dir =  os.path.abspath(dst_path)
+        else:
+            local_dir = _tmp_dir(dst_path)
 
         with hdfs_system(scheme=self.scheme, host=self.host, port=self.port) as hdfs: