Tweaks

openvinotoolkit · Nov 14, 2024 · e7617f1 · e7617f1
1 parent 5f46593
commit e7617f1
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/tests/openvino/native/quantization/test_ov_modeling_compression.py b/tests/openvino/native/quantization/test_ov_modeling_compression.py
@@ -99,12 +99,12 @@ def get_random_integer_tensor(shape, low, high, dtype, backend, seed=0):
 
 @contextmanager
 def openvino_available(available: bool):
-    import nncf.utils
+    import nncf.import_utils
 
-    original_value = nncf.utils._openvino_available
-    nncf.utils._openvino_available = available
+    original_value = nncf.import_utils._openvino_available
+    nncf.import_utils._openvino_available = available
     yield
-    nncf.utils._openvino_available = original_value
+    nncf.import_utils._openvino_available = original_value
 
 
 @pytest.mark.parametrize("weight_shape", [(10000, 4)], ids=[""])