Skip to content

Commit 5559721

Browse files
author
Richard He
committed
Do not raise when model version path is s3
Signed-off-by: Richard He <richard@graft.com>
1 parent afe3940 commit 5559721

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

mlflow_export_import/model/import_model.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -91,7 +91,7 @@ def _import_version(self,
9191
:param sleep_time: Seconds to wait for model version crreation.
9292
"""
9393
dst_source = dst_source.replace("file://","") # OSS MLflow
94-
if not dst_source.startswith("dbfs:") and not os.path.exists(dst_source):
94+
if not dst_source.startswith("dbfs:") and not dst_source.startswith("s3:") and not os.path.exists(dst_source):
9595
raise MlflowExportImportException(f"'source' argument for MLflowClient.create_model_version does not exist: {dst_source}", http_status_code=404)
9696
kwargs = {"await_creation_for": self.await_creation_for } if self.await_creation_for else {}
9797
tags = src_vr["tags"]

0 commit comments

Comments
 (0)