From 72b9a8bbda2e1c6115c63c69ca43897029ebf84f Mon Sep 17 00:00:00 2001 From: lamttic Date: Sun, 24 Dec 2023 01:27:51 +0900 Subject: [PATCH] Create temporary directory using uuid --- model-archiver/model_archiver/model_packaging_utils.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/model-archiver/model_archiver/model_packaging_utils.py b/model-archiver/model_archiver/model_packaging_utils.py index 01ac568a95..f153bcb82c 100644 --- a/model-archiver/model_archiver/model_packaging_utils.py +++ b/model-archiver/model_archiver/model_packaging_utils.py @@ -9,6 +9,7 @@ import shutil import tarfile import tempfile +import uuid import zipfile from io import BytesIO from pathlib import Path @@ -152,7 +153,7 @@ def copy_artifacts(model_name, **kwargs): :param kwargs: key value pair of files to be copied in archive :return: """ - model_path = os.path.join(tempfile.gettempdir(), model_name) + model_path = os.path.join(tempfile.gettempdir(), model_name, uuid.uuid4().hex) if os.path.exists(model_path): shutil.rmtree(model_path) ModelExportUtils.make_dir(model_path)