From fb2e858fcedc1baf5fc1c2f67c4eceb179559f94 Mon Sep 17 00:00:00 2001 From: Aleksei Kashapov Date: Thu, 14 Nov 2024 13:13:27 +0100 Subject: [PATCH] upd to RawReducer --- .../test_templates/test_weights_compression_backends.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/cross_fw/test_templates/test_weights_compression_backends.py b/tests/cross_fw/test_templates/test_weights_compression_backends.py index fb521fd0726..4d72e62239a 100644 --- a/tests/cross_fw/test_templates/test_weights_compression_backends.py +++ b/tests/cross_fw/test_templates/test_weights_compression_backends.py @@ -18,7 +18,7 @@ from nncf.experimental.common.tensor_statistics.collectors import MeanAbsMaxReducer from nncf.experimental.common.tensor_statistics.collectors import MeanAggregator from nncf.experimental.common.tensor_statistics.collectors import MeanVarianceReducer -from nncf.experimental.common.tensor_statistics.collectors import NoopReducer +from nncf.experimental.common.tensor_statistics.collectors import RawReducer from nncf.experimental.common.tensor_statistics.collectors import TensorCollector @@ -58,7 +58,7 @@ def check_reducer(self, collector: TensorCollector, expected_reducer_type): @pytest.mark.parametrize( "algo_func, aggregator_type, reducer_type", [ - ("get_hawq_with_backend", HAWQAggregator, NoopReducer), + ("get_hawq_with_backend", HAWQAggregator, RawReducer), ("get_mean_variance_with_backend", MeanAggregator, MeanVarianceReducer), ("get_max_variance_with_backend", MeanAggregator, MaxVarianceReducer), ("get_mean_max_with_backend", MeanAggregator, MeanAbsMaxReducer),