fix ruff

Author: l-bat

optimum/intel/openvino/modeling_diffusion.py

@@ -35,6 +35,7 @@
 from diffusers.schedulers.scheduling_utils import SCHEDULER_CONFIG_NAME
 from diffusers.utils import CONFIG_NAME, is_invisible_watermark_available
 from huggingface_hub import snapshot_download
+from nncf import Dataset
 from openvino._offline_transformations import compress_model_transformation
 from openvino.runtime import Core
 from transformers import CLIPFeatureExtractor, CLIPTokenizer
@@ -341,13 +342,13 @@ def _from_pretrained(
 
     def prepare_inputs(
         self,
-        dataset: "Dataset",
+        dataset: Dataset,
         subset_size: int,
         num_inference_steps: int,
         height: Optional[int] = 512,
         width: Optional[int] = 512,
         **kwargs,
-    ) -> "Dataset":
+    ) -> Dataset:
         self.compile()
         calibration_data = []
 
@@ -359,9 +360,6 @@ def prepare_inputs(
             if len(calibration_data) >= subset_size:
                 break
         self.unet.request = self.unet.request.request
-
-        from nncf import Dataset
-
         return Dataset(calibration_data)
 
     @classmethod

optimum/intel/openvino/quantization.py

@@ -18,14 +18,14 @@
 import os
 from collections import deque
 from copy import deepcopy
-from datasets import load_dataset
 from pathlib import Path
 from typing import TYPE_CHECKING, Any, Callable, Dict, Optional, Tuple, Union
 
 import nncf
 import openvino
 import torch
 import transformers
+from datasets import load_dataset
 from nncf import CompressWeightsMode, IgnoredScope, NNCFConfig, SensitivityMetric
 from nncf.quantization.advanced_parameters import AdvancedSmoothQuantParameters
 from nncf.torch import create_compressed_model, register_default_init_args, register_module
@@ -588,7 +588,7 @@ def _weight_only_quantization(
         # awq=config.quant_method == "awq", # TODO : remove and add it back once nncf v2.9.0
         ignored_scope=ignored_scope,
         dataset=dataset,
-        subset_size=config.subset_size,
+        # subset_size=config.subset_size, # TODO : enable from nncf v2.9.0
     )