make torch precommit green

openvinotoolkit · Jul 29, 2023 · 4e00871 · 4e00871
1 parent 2d1403e
commit 4e00871
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git 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
@@ -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": {