Fix ruff

nikita-savelyevv · nikita-savelyevv · commit f314ba06b32e · 2024-04-15T09:22:19.000+02:00
diff --git a/optimum/intel/openvino/modeling_base.py b/optimum/intel/openvino/modeling_base.py
@@ -103,7 +103,7 @@ def __init__(
     def load_model(
         file_name: Union[str, Path],
         quantization_config: Union[OVWeightQuantizationConfig, Dict] = None,
-        calibration_dataset: Optional["nncf.Dataset"] = None,
+        calibration_dataset: Optional["Dataset"] = None,
     ):
         """
         Loads the model.
