From 920a39eee28fff25eb73f0f4a3f705bf2c2ff935 Mon Sep 17 00:00:00 2001 From: Michael Feil Date: Fri, 19 May 2023 01:34:43 +0200 Subject: [PATCH] fix tests --- tests/test_translate.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/test_translate.py b/tests/test_translate.py index 43cbbbf..c832637 100644 --- 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(