bug fix

Signed-off-by: chensuyue <[email protected]>

Author: chensuyue

neural_compressor/torch/utils/environ.py

@@ -46,6 +46,7 @@ def is_package_available(package_name):
 ## check hpex
 if is_package_available("habana_frameworks"):
     _hpex_available = True
+    import habana_frameworks.torch.hpex  # pylint: disable=E0401
 else:
     _hpex_available = False
 
@@ -57,7 +58,6 @@ def is_hpex_available():
 ## check ipex
 if is_package_available("intel_extension_for_pytorch"):
     _ipex_available = True
-    import habana_frameworks.torch.hpex  # pylint: disable=E0401
 else:
     _ipex_available = False