diff --git a/tests/torch/ptq/test_calculation_quantizer_params.py b/tests/torch/ptq/test_calculation_quantizer_params.py index cec246f1ad7..f98e5137d48 100644 --- a/tests/torch/ptq/test_calculation_quantizer_params.py +++ b/tests/torch/ptq/test_calculation_quantizer_params.py @@ -318,7 +318,9 @@ def test_quantizer_parameters_export(tmp_path: Path): statistic_points = min_max_algo.get_statistic_points(nncf_network, nncf_network.nncf.get_graph()) statistics_aggregator.register_statistic_points(statistic_points) statistics_aggregator.collect_statistics(model, nncf_network.nncf.get_graph()) - torch_quantized_model = min_max_algo.apply(model, statistics_aggregator.statistic_points) + torch_quantized_model = min_max_algo.apply( + nncf_network, nncf_network.nncf.get_graph(), statistics_aggregator.statistic_points + ) path = str(tmp_path / "torch_ptq_model.onnx") torch.onnx.export( diff --git a/tests/torch/ptq/test_ptq_params.py b/tests/torch/ptq/test_ptq_params.py index 7d657d81441..c174ec8b322 100644 --- a/tests/torch/ptq/test_ptq_params.py +++ b/tests/torch/ptq/test_ptq_params.py @@ -141,7 +141,7 @@ def test_params(self): }, "test_range_estimator_per_channel": { "model": depthwise_model, - "nncf_graph": depthwise_model.nncf.graph, + "nncf_graph": depthwise_model.nncf.get_graph(), "stat_points_num": 2, }, "test_quantize_outputs": {