Fix command in xgboost example readme (mlflow#5202)

Signed-off-by: Junwen Yao <jwyiao@gmail.com> Signed-off-by: Dima Claudiu <claudiu.dima@soleadify.com>
DimaClaudiu · Dec 30, 2021 · 046a996 · 046a996
1 parent 4c657b3
commit 046a996
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/examples/xgboost/xgboost_sklearn/README.md b/examples/xgboost/xgboost_sklearn/README.md
@@ -5,8 +5,9 @@ This example trains an [`XGBoost.XGBRegressor`](https://xgboost.readthedocs.io/e
 Like the other XGBoost example, we enable autologging for XGBoost scikit-learn models via `mlflow.xgboost.autolog()`. Saving / loading models also supports XGBoost scikit-learn models.
 
 You can run this example using the following command:
+
 ```
-python train_sklearn.py
+python train.py
 ```
 
 
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: