Skip to content

Commit 84c6f50

Browse files
committed
update tests
1 parent ada7dba commit 84c6f50

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/test_translate.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
from hf_hub_ctranslate2 import TranslatorCT2fromHfHub, GeneratorCT2fromHfHub, MultiLingualTranslatorCT2fromHfHub
2-
from hf_hub_ctranslate2.util import _download_model
2+
from hf_hub_ctranslate2.util import utils as _utils
33
from transformers import AutoTokenizer
44

55

@@ -51,7 +51,7 @@ def test_generator(model_name="michaelfeil/ct2fast-pythia-160m"):
5151

5252

5353
def test_generator_single(model_name="michaelfeil/ct2fast-pythia-160m"):
54-
model_path = _download_model(model_name)
54+
model_path = _utils._download_model(model_name)
5555
tokenizer = AutoTokenizer.from_pretrained(model_path)
5656

5757
model = GeneratorCT2fromHfHub(

0 commit comments

Comments
 (0)