Fix imports 6

Author: nikita-savelyevv

optimum/intel/openvino/quantization/calibration_dataset_builder.py

@@ -16,11 +16,13 @@
 import logging
 from typing import Any, Callable, Dict, List, Optional, Sized, Tuple, Union
 
+import nncf
 import openvino
 import requests
 import torch
 import transformers
 from huggingface_hub.constants import HUGGINGFACE_HUB_CACHE
+from nncf.torch.initialization import PTInitializingDataLoader
 from PIL import Image
 from torch.utils.data import DataLoader, RandomSampler
 from tqdm import tqdm
@@ -37,7 +39,6 @@
     is_datasets_available,
     is_datasets_version,
     is_diffusers_available,
-    is_nncf_available,
 )
 
 from .configuration import OVQuantizationConfigBase
@@ -46,10 +47,6 @@
 if is_datasets_available():
     from datasets import Dataset
 
-if is_nncf_available():
-    import nncf
-    from nncf.torch.initialization import PTInitializingDataLoader
-
 logger = logging.getLogger(__name__)
 
 

optimum/intel/openvino/quantization/quantizer.py

@@ -48,7 +48,6 @@
     ONNX_WEIGHTS_NAME,
     OV_XML_FILE_NAME,
 )
-from .calibration_dataset_builder import OVCalibrationDatasetBuilder
 from .configuration import (
     OVConfig,
     OVMixedQuantizationConfig,
@@ -66,6 +65,8 @@
     from nncf.quantization.advanced_parameters import OverflowFix
     from nncf.torch import register_module
 
+    from .calibration_dataset_builder import OVCalibrationDatasetBuilder
+
 register_module(ignored_algorithms=[])(Conv1D)
 
 core = Core()
@@ -90,6 +91,11 @@ def __init__(self, model: transformers.PreTrainedModel, task: Optional[str] = No
         super().__init__()
         self.model = model
         self.task = task
+
+        if not is_nncf_available():
+            # TODO: add message
+            raise Exception
+
         self.dataset_builder = OVCalibrationDatasetBuilder(model, seed)
 
     @classmethod