fix tests

michaelfeil · May 18, 2023 · 920a39e · 920a39e
1 parent 59800dd
commit 920a39e
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
-import hf_hub_ctranslate2.util.utils as utils
+from hf_hub_ctranslate2.util import _download_model
 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 = utils.download_model(model_name)
+    model_path = _download_model(model_name)
     tokenizer = AutoTokenizer.from_pretrained(model_path)
 
     model = GeneratorCT2fromHfHub(