From 5e859c2e7b0f828ae8ba218e32d680dc29221990 Mon Sep 17 00:00:00 2001 From: dlyakhov Date: Wed, 27 Sep 2023 15:00:57 +0200 Subject: [PATCH] Minor --- tests/common/test_statistics_aggregator.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/common/test_statistics_aggregator.py b/tests/common/test_statistics_aggregator.py index 6bfcf9ef753..6c5fc043904 100644 --- a/tests/common/test_statistics_aggregator.py +++ b/tests/common/test_statistics_aggregator.py @@ -830,7 +830,7 @@ def test_same_collectors_different_attrs_dont_merge(self, statistics_type, test_ elif statistics_type == "mean_per_ch": # TODO(dlyakhov) uncoment when nncf will switch to openvino==2023.0.0 # params["inplace"] = [False, True] - params["channel_dim"] = [1, 2] + params["channel_axis"] = [1, 2] def product_dict(**kwargs): keys = kwargs.keys() @@ -852,7 +852,7 @@ def product_dict(**kwargs): dataset = self.get_dataset(dataset_samples) statistics_aggregator = self.get_statistics_aggregator(dataset) statistics_aggregator.register_statistic_points(statistics_points) - # Run statistic collection to check output names matches reduer names + # Run statistic collection to check output names matches reducer names graph = NNCFGraphFactory.create(model) statistics_aggregator.collect_statistics(model, graph)