Use mkstemp to replace deprecated mktemp call (#5303)

* Use mkstemp Signed-off-by: dbczumar <corey.zumar@databricks.com> * Remove num examples Signed-off-by: dbczumar <corey.zumar@databricks.com> * Close instead of remove Signed-off-by: dbczumar <corey.zumar@databricks.com> * Close the handle Signed-off-by: dbczumar <corey.zumar@databricks.com>
mlflow · Jan 26, 2022 · 61984e6 · 61984e6 · sr-mpamera · Mar 2, 2022
1 parent 271750b
commit 61984e6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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=""):