Commit e5019b2f authored by Neel Shah's avatar Neel Shah
Browse files

minor bug fix, allowing spaces to be an integer and not just a tuple with a single integer

parent f3e0cd1a
Pipeline #106602 failed with stages
......@@ -134,7 +134,7 @@ class MatrixProductOperator(LinearOperator):
active_axes = []
self._inactive_axes = list(range(len(self._domain)))
domain_shape = []
for space_idx in spaces:
for space_idx in self._spaces:
domain_shape += self._domain[space_idx].shape
active_axes += self._domain.axes[space_idx]
if space_idx in self._inactive_axes:
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment