Skip to content

Commit aa183ff

Browse files
Black
1 parent 7754721 commit aa183ff

File tree

2 files changed

+10
-6
lines changed

2 files changed

+10
-6
lines changed

optimum/intel/openvino/quantization.py

+7-4
Original file line numberDiff line numberDiff line change
@@ -86,10 +86,13 @@ class InferRequestWrapper:
8686
Wrapper class for OV InferRequest or CompiledModel objects that collects inputs which they were called with to
8787
a list.
8888
"""
89-
def __init__(self,
90-
request: Union[openvino.InferRequest, openvino.CompiledModel],
91-
collected_inputs: List = None,
92-
apply_caching: bool = False):
89+
90+
def __init__(
91+
self,
92+
request: Union[openvino.InferRequest, openvino.CompiledModel],
93+
collected_inputs: List = None,
94+
apply_caching: bool = False,
95+
):
9396
"""
9497
Args:
9598
request (`Union[openvino.InferRequest, openvino.CompiledModel]`):

tests/openvino/test_quantization.py

+3-2
Original file line numberDiff line numberDiff line change
@@ -696,8 +696,9 @@ def test_calibration_data_uniqueness(self, model_id, apply_caching):
696696
processor = AutoProcessor.from_pretrained(model_id)
697697

698698
calibration_data = []
699-
ov_model.decoder_with_past.request = InferRequestWrapper(ov_model.decoder_with_past.request, calibration_data,
700-
apply_caching=apply_caching)
699+
ov_model.decoder_with_past.request = InferRequestWrapper(
700+
ov_model.decoder_with_past.request, calibration_data, apply_caching=apply_caching
701+
)
701702
for _ in range(2):
702703
input_features = self._generate_random_audio_data(processor)
703704
ov_model.generate(input_features)

0 commit comments

Comments
 (0)