diff --git a/mlflow_export_import/model/import_model.py b/mlflow_export_import/model/import_model.py index 9c3957cb..cc3352c7 100644 --- a/mlflow_export_import/model/import_model.py +++ b/mlflow_export_import/model/import_model.py @@ -91,7 +91,7 @@ def _import_version(self, :param sleep_time: Seconds to wait for model version crreation. """ dst_source = dst_source.replace("file://","") # OSS MLflow - if not dst_source.startswith("dbfs:") and not os.path.exists(dst_source): + if not dst_source.startswith("dbfs:") and not dst_source.startswith("s3:") and not os.path.exists(dst_source): raise MlflowExportImportException(f"'source' argument for MLflowClient.create_model_version does not exist: {dst_source}", http_status_code=404) kwargs = {"await_creation_for": self.await_creation_for } if self.await_creation_for else {} tags = src_vr["tags"]