add fixed import with local import

michaelfeil · May 18, 2023 · db71851 · db71851
1 parent 372b610
commit db71851
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 5 deletions.
diff --git a/hf_hub_ctranslate2/__init__.py b/hf_hub_ctranslate2/__init__.py
@@ -1,6 +1,6 @@
 # -*- coding: utf-8 -*-
 """Compatability between Huggingface and Ctranslate2."""
-__all__ = ["__version__", "TranslatorCT2fromHfHub", "GeneratorCT2fromHfHub", "MultiLingualTranslatorCT2fromHfHub", "_private"]
+# __all__ = ["__version__", "TranslatorCT2fromHfHub", "GeneratorCT2fromHfHub", "MultiLingualTranslatorCT2fromHfHub", "_private"]
 from hf_hub_ctranslate2.translate import TranslatorCT2fromHfHub, GeneratorCT2fromHfHub, MultiLingualTranslatorCT2fromHfHub
-import hf_hub_ctranslate2._private as _private
+
 __version__ = "2.0.2"
diff --git a/hf_hub_ctranslate2/_private/__init__.py b/hf_hub_ctranslate2/_private/__init__.py
@@ -0,0 +1,2 @@
+__all__ = ["download_model"]
+from .utils import download_model
diff --git a/hf_hub_ctranslate2/translate.py b/hf_hub_ctranslate2/translate.py
@@ -14,7 +14,7 @@
 from typing import Any, Union, List
 import os
 
-import hf_hub_ctranslate2._private.utils as _utils
+from ._private import download_model as _download_model
 
 
 class CTranslate2ModelfromHuggingfaceHub:
@@ -34,10 +34,10 @@ def __init__(
             model_path = model_name_or_path
         else:
             try:
-                model_path = _utils.download_model(model_name_or_path, hub_kwargs=hub_kwargs)
+                model_path = _download_model(model_name_or_path, hub_kwargs=hub_kwargs)
             except:
                 hub_kwargs["local_files_only"] = True
-                model_path = _utils.download_model(model_name_or_path, hub_kwargs=hub_kwargs)
+                model_path = _download_model(model_name_or_path, hub_kwargs=hub_kwargs)
         self.model = self.ctranslate_class(
             model_path,
             device=device,
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1,2 @@
		__all__ = ["download_model"]
		from .utils import download_model