update tests

michaelfeil · May 20, 2023 · 84c6f50 · 84c6f50
1 parent ada7dba
commit 84c6f50
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_translate.py b/tests/test_translate.py
@@ -1,5 +1,5 @@
 from hf_hub_ctranslate2 import TranslatorCT2fromHfHub, GeneratorCT2fromHfHub, MultiLingualTranslatorCT2fromHfHub
-from hf_hub_ctranslate2.util import _download_model
+from hf_hub_ctranslate2.util import utils as _utils
 from transformers import AutoTokenizer
 
 
@@ -51,7 +51,7 @@ def test_generator(model_name="michaelfeil/ct2fast-pythia-160m"):
 
 
 def test_generator_single(model_name="michaelfeil/ct2fast-pythia-160m"):
-    model_path = _download_model(model_name)
+    model_path = _utils._download_model(model_name)
     tokenizer = AutoTokenizer.from_pretrained(model_path)
 
     model = GeneratorCT2fromHfHub(