From df2ef398cda5392f7f2c6474eadaa7cb489e61dd Mon Sep 17 00:00:00 2001 From: Tim Huang Date: Thu, 29 Apr 2021 14:38:57 +0800 Subject: [PATCH] bug fix for missing artifacts in imported runs from DBFS --- mlflow_export_import/run/import_run.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mlflow_export_import/run/import_run.py b/mlflow_export_import/run/import_run.py index 417656f..8600170 100644 --- a/mlflow_export_import/run/import_run.py +++ b/mlflow_export_import/run/import_run.py @@ -47,7 +47,7 @@ def import_run_from_dir(self, dst_exp_name, src_run_id): run_id = run.info.run_id self.import_run_data(src_run_dct, run_id, src_run_dct["info"]["user_id"]) path = os.path.join(src_run_id,"artifacts") - if os.path.exists(path): + if os.path.exists(mk_local_path(path)): mlflow.log_artifacts(mk_local_path(path)) if self.mlmodel_fix: self.update_mlmodel_run_id(run.info.run_id)