From 88f51fe7d2528394d11a82d5ca0e5db50d99a89b Mon Sep 17 00:00:00 2001 From: Lucain Date: Tue, 11 Jun 2024 15:26:58 +0200 Subject: [PATCH] Switch from cached_download to hf_hub_download in tests (#1547) --- bindings/python/tests/test_serialization.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/bindings/python/tests/test_serialization.py b/bindings/python/tests/test_serialization.py index a56c6bb3..4434e630 100644 --- a/bindings/python/tests/test_serialization.py +++ b/bindings/python/tests/test_serialization.py @@ -3,9 +3,9 @@ import os import unittest import tqdm - -from huggingface_hub import HfApi, cached_download, hf_hub_url +from huggingface_hub import hf_hub_download from tokenizers import Tokenizer + from .utils import albert_base, data_dir @@ -63,7 +63,7 @@ class TestFullDeserialization(unittest.TestCase): all_models = [("HueyNemud/das22-10-camembert_pretrained", "tokenizer.json")] for model_id, filename in tqdm.tqdm(all_models): - tokenizer_file = cached_download(hf_hub_url(model_id, filename=filename)) + tokenizer_file = hf_hub_download(model_id, filename=filename) is_ok = check(tokenizer_file) if not is_ok: