diff --git a/optimum/habana/utils/__init__.py b/optimum/habana/utils/__init__.py new file mode 100644 index 0000000000..5bade17cdf --- /dev/null +++ b/optimum/habana/utils/__init__.py @@ -0,0 +1 @@ +from .misc import * # NOTE(pbielak): mimic previous imports from `utils.py` diff --git a/optimum/habana/utils.py b/optimum/habana/utils/misc.py similarity index 99% rename from optimum/habana/utils.py rename to optimum/habana/utils/misc.py index 74f42f909b..4239fabebc 100755 --- a/optimum/habana/utils.py +++ b/optimum/habana/utils/misc.py @@ -26,7 +26,7 @@ from optimum.utils import logging -from .version import __version__ +from ..version import __version__ logger = logging.get_logger(__name__)