Skip to content

Commit 0eb567b

Browse files
Fix test after merging
1 parent 6a31129 commit 0eb567b

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

tests/openvino/test_quantization.py

+4-4
Original file line numberDiff line numberDiff line change
@@ -1394,16 +1394,16 @@ def test_config_init_kwargs(
13941394

13951395
@parameterized.expand(
13961396
[
1397-
("nncf.compress_weights", "_weight_only_quantization", "dataset"),
1398-
("nncf.quantize", "_full_quantization", "calibration_dataset"),
1397+
("nncf.compress_weights", "_weight_only_quantization", "dataset", OVWeightQuantizationConfig),
1398+
("nncf.quantize", "_full_quantization", "calibration_dataset", OVQuantizationConfig),
13991399
]
14001400
)
1401-
def test_quantization_kwargs_override(self, mock_method_name, quantization_function, dataset_key):
1401+
def test_quantization_kwargs_override(self, mock_method_name, quantization_function, dataset_key, config_type):
14021402
with unittest.mock.patch(mock_method_name) as mock_method:
14031403
mock_model = unittest.mock.Mock([])
14041404
mock_model.get_rt_info = unittest.mock.Mock(return_value={})
14051405

1406-
mock_quantization_config = unittest.mock.Mock()
1406+
mock_quantization_config = unittest.mock.Mock(config_type)
14071407
mock_quantization_config.to_nncf_dict.return_value = {"param1": "value1", "param2": "value2"}
14081408

14091409
additional_kwargs = {"param2": "new_value2", "param3": "value3"}

0 commit comments

Comments
 (0)