diff --git a/mlflow/utils/file_utils.py b/mlflow/utils/file_utils.py index 32653b343c445..4b59a367372af 100644 --- a/mlflow/utils/file_utils.py +++ b/mlflow/utils/file_utils.py @@ -287,7 +287,7 @@ def _filter_timestamps(tar_info): tar_info.mtime = 0 return tar_info if custom_filter is None else custom_filter(tar_info) - unzipped_filename = tempfile.mktemp() + unzipped_file_handle, unzipped_filename = tempfile.mkstemp() try: with tarfile.open(unzipped_filename, "w") as tar: tar.add(source_dir, arcname=archive_name, filter=_filter_timestamps) @@ -298,7 +298,7 @@ def _filter_timestamps(tar_info): ) as gzipped_tar, open(unzipped_filename, "rb") as tar: gzipped_tar.write(tar.read()) finally: - os.remove(unzipped_filename) + os.close(unzipped_file_handle) def _copy_project(src_path, dst_path=""):