Skip to content

Commit

Permalink
Minor
Browse files Browse the repository at this point in the history
  • Loading branch information
daniil-lyakhov committed Sep 27, 2023
1 parent 43536c4 commit 5e859c2
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions tests/common/test_statistics_aggregator.py
Original file line number Diff line number Diff line change
Expand Up @@ -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()
Expand All @@ -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)

Expand Down

0 comments on commit 5e859c2

Please sign in to comment.