Skip to content

Commit 2664489

Browse files
author
Nikita Savelyev
committed
Fix condition
1 parent 04a05b5 commit 2664489

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/nncf/experimental/common/tensor_statistics/statistics.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -110,8 +110,8 @@ class MeanTensorStatistic(TensorStatistic):
110110
shape: tuple[int, ...]
111111

112112
def __post_init__(self):
113-
if isinstance(self.shape, (deque, list)):
114-
# If NoopAggregator was used, shape can be a sequence containing a single tensor
113+
if isinstance(self.shape[0], Tensor):
114+
# If in-place shape reducer and Noop aggregator were used, shape is a sequence containing a single tensor
115115
self.shape = tuple(self.shape[0].data.tolist())
116116

117117
def __eq__(self, other: TensorStatistic):

0 commit comments

Comments
 (0)