diff --git a/linear_operator/operators/block_linear_operator.py b/linear_operator/operators/block_linear_operator.py index 8ad44763..54862816 100644 --- a/linear_operator/operators/block_linear_operator.py +++ b/linear_operator/operators/block_linear_operator.py @@ -97,7 +97,7 @@ def _getitem(self, row_index: IndexType, col_index: IndexType, *batch_indices: I col_index = slice(col_start // num_blocks, col_end // num_blocks, None) # Now we can try the super call! - new_base_linear_op = self.base_linear_op._getitem(row_index, col_index, *batch_indices) + new_base_linear_op = self.base_linear_op._getitem(row_index, col_index, *batch_indices, _noop_index) # Now construct a kernel with those indices return self.__class__(new_base_linear_op, block_dim=-3)