Skip to content

Commit a33f348

Browse files
use _prepare_artifact_dir for temp directory creation
1 parent 14c4856 commit a33f348

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

ads/model/generic_model.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1448,9 +1448,9 @@ def download_artifact(
14481448
artifact_dir = _prepare_artifact_dir(artifact_dir)
14491449

14501450
target_dir = (
1451-
artifact_dir
1452-
if not ObjectStorageDetails.is_oci_path(artifact_dir)
1453-
else tempfile.mkdtemp()
1451+
_prepare_artifact_dir()
1452+
if ObjectStorageDetails.is_oci_path(artifact_dir)
1453+
else artifact_dir
14541454
)
14551455

14561456
dsc_model = DataScienceModel.from_id(model_id)
@@ -1585,9 +1585,9 @@ def from_model_catalog(
15851585
artifact_dir = _prepare_artifact_dir(artifact_dir)
15861586

15871587
target_dir = (
1588-
artifact_dir
1589-
if not ObjectStorageDetails.is_oci_path(artifact_dir)
1590-
else tempfile.mkdtemp()
1588+
_prepare_artifact_dir()
1589+
if ObjectStorageDetails.is_oci_path(artifact_dir)
1590+
else artifact_dir
15911591
)
15921592
bucket_uri = bucket_uri or (
15931593
artifact_dir if ObjectStorageDetails.is_oci_path(artifact_dir) else None

0 commit comments

Comments
 (0)