Skip to content

Commit dbed373

Browse files
authored
Merge pull request #115 from LeoIV/main
Adding _noop_index for the block dimension in block_linear_operator.py
2 parents d8052c9 + 9fc4c26 commit dbed373

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

linear_operator/operators/block_linear_operator.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -97,7 +97,7 @@ def _getitem(self, row_index: IndexType, col_index: IndexType, *batch_indices: I
9797
col_index = slice(col_start // num_blocks, col_end // num_blocks, None)
9898

9999
# Now we can try the super call!
100-
new_base_linear_op = self.base_linear_op._getitem(row_index, col_index, *batch_indices)
100+
new_base_linear_op = self.base_linear_op._getitem(row_index, col_index, *batch_indices, _noop_index)
101101

102102
# Now construct a kernel with those indices
103103
return self.__class__(new_base_linear_op, block_dim=-3)

0 commit comments

Comments
 (0)