Black

Author: nikita-savelyevv

optimum/intel/openvino/quantization.py

@@ -86,10 +86,13 @@ class InferRequestWrapper:
     Wrapper class for OV InferRequest or CompiledModel objects that collects inputs which they were called with to
     a list.
     """
-    def __init__(self,
-                 request: Union[openvino.InferRequest, openvino.CompiledModel],
-                 collected_inputs: List = None,
-                 apply_caching: bool = False):
+
+    def __init__(
+        self,
+        request: Union[openvino.InferRequest, openvino.CompiledModel],
+        collected_inputs: List = None,
+        apply_caching: bool = False,
+    ):
         """
         Args:
             request (`Union[openvino.InferRequest, openvino.CompiledModel]`):

tests/openvino/test_quantization.py

@@ -696,8 +696,9 @@ def test_calibration_data_uniqueness(self, model_id, apply_caching):
         processor = AutoProcessor.from_pretrained(model_id)
 
         calibration_data = []
-        ov_model.decoder_with_past.request = InferRequestWrapper(ov_model.decoder_with_past.request, calibration_data,
-                                                                 apply_caching=apply_caching)
+        ov_model.decoder_with_past.request = InferRequestWrapper(
+            ov_model.decoder_with_past.request, calibration_data, apply_caching=apply_caching
+        )
         for _ in range(2):
             input_features = self._generate_random_audio_data(processor)
             ov_model.generate(input_features)