From 0fcfc5ccc968ff5a1a439db04a94f566a0bd1d89 Mon Sep 17 00:00:00 2001 From: Sai-Suraj-27 Date: Mon, 22 Jul 2024 18:43:39 +0530 Subject: [PATCH] fix: Replaced deprecated `mktemp()` function (#32123) Replaced deprecated mktemp function. --- tests/utils/test_image_utils.py | 6 +++--- tests/utils/test_tokenization_utils.py | 2 +- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/tests/utils/test_image_utils.py b/tests/utils/test_image_utils.py index f360c4bb82..1fa84aa5db 100644 --- a/tests/utils/test_image_utils.py +++ b/tests/utils/test_image_utils.py @@ -513,7 +513,7 @@ class LoadImageTester(unittest.TestCase): 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 @@ class LoadImageTester(unittest.TestCase): 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 @@ class LoadImageTester(unittest.TestCase): 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 index 0df86dc3cc..929ee5260a 100644 --- a/tests/utils/test_tokenization_utils.py +++ b/tests/utils/test_tokenization_utils.py @@ -84,7 +84,7 @@ class TokenizerUtilTester(unittest.TestCase): 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)