Skip to content

Commit

Permalink
Switch from cached_download to hf_hub_download in tests (#1547)
Browse files Browse the repository at this point in the history
  • Loading branch information
Wauplin authored Jun 11, 2024
1 parent 418c35c commit 88f51fe
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions bindings/python/tests/test_serialization.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,9 +3,9 @@
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


Expand Down Expand Up @@ -63,7 +63,7 @@ def test_full_deserialization_hub(self):

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:
Expand Down

0 comments on commit 88f51fe

Please sign in to comment.