fix: Replaced deprecated mktemp() function (huggingface#32123)

Replaced deprecated mktemp function.
MHRDYN7 · Jul 23, 2024 · 53c534f · 53c534f
1 parent 84d5476
commit 53c534f
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/tests/utils/test_image_utils.py b/tests/utils/test_image_utils.py
@@ -513,7 +513,7 @@ def test_load_img_local(self):
 
     def test_load_img_base64_prefix(self):
         try:
-            tmp_file = tempfile.mktemp()
+            tmp_file = tempfile.NamedTemporaryFile(delete=False).name
             with open(tmp_file, "wb") as f:
                 http_get(
                     "https://huggingface.co/datasets/hf-internal-testing/dummy-base64-images/raw/main/image_0.txt", f
@@ -530,7 +530,7 @@ def test_load_img_base64_prefix(self):
 
     def test_load_img_base64(self):
         try:
-            tmp_file = tempfile.mktemp()
+            tmp_file = tempfile.NamedTemporaryFile(delete=False).name
             with open(tmp_file, "wb") as f:
                 http_get(
                     "https://huggingface.co/datasets/hf-internal-testing/dummy-base64-images/raw/main/image_1.txt", f
@@ -547,7 +547,7 @@ def test_load_img_base64(self):
 
     def test_load_img_base64_encoded_bytes(self):
         try:
-            tmp_file = tempfile.mktemp()
+            tmp_file = tempfile.NamedTemporaryFile(delete=False).name
             with open(tmp_file, "wb") as f:
                 http_get(
                     "https://huggingface.co/datasets/hf-internal-testing/dummy-base64-images/raw/main/image_2.txt", f

diff --git a/tests/utils/test_tokenization_utils.py b/tests/utils/test_tokenization_utils.py
@@ -84,7 +84,7 @@ def test_cached_files_are_used_when_internet_is_down_missing_files(self):
     def test_legacy_load_from_one_file(self):
         # This test is for deprecated behavior and can be removed in v5
         try:
-            tmp_file = tempfile.mktemp()
+            tmp_file = tempfile.NamedTemporaryFile(delete=False).name
             with open(tmp_file, "wb") as f:
                 http_get("https://huggingface.co/albert/albert-base-v1/resolve/main/spiece.model", f)