SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

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

Browse Source 
Replaced deprecated mktemp function.
This commit is contained in:
Sai-Suraj-27
2024-07-22 18:43:39 +05:30
committed by GitHub
parent c38c55f4fb
commit 0fcfc5ccc9
2 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tests/utils/test_image_utils.py
View File

@@ -513,7 +513,7 @@ class LoadImageTester(unittest.TestCase):
def test_load_img_base64_prefix(self): def test_load_img_base64_prefix(self):
try: try:
tmp_file = tempfile.mktemp() tmp_file = tempfile.NamedTemporaryFile(delete=False).name
with open(tmp_file, "wb") as f: with open(tmp_file, "wb") as f:
http_get( http_get(
"https://huggingface.co/datasets/hf-internal-testing/dummy-base64-images/raw/main/image_0.txt", f "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): def test_load_img_base64(self):
try: try:
tmp_file = tempfile.mktemp() tmp_file = tempfile.NamedTemporaryFile(delete=False).name
with open(tmp_file, "wb") as f: with open(tmp_file, "wb") as f:
http_get( http_get(
"https://huggingface.co/datasets/hf-internal-testing/dummy-base64-images/raw/main/image_1.txt", f "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): def test_load_img_base64_encoded_bytes(self):
try: try:
tmp_file = tempfile.mktemp() tmp_file = tempfile.NamedTemporaryFile(delete=False).name
with open(tmp_file, "wb") as f: with open(tmp_file, "wb") as f:
http_get( http_get(
"https://huggingface.co/datasets/hf-internal-testing/dummy-base64-images/raw/main/image_2.txt", f "https://huggingface.co/datasets/hf-internal-testing/dummy-base64-images/raw/main/image_2.txt", f

2
tests/utils/test_tokenization_utils.py
View File

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