Skip to content
GitLab
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
Hide whitespace changes
Inline Side-by-side
src/operators/matrix_product_operator.py
View file @ e5019b2f
...@@ -134,7 +134,7 @@ class MatrixProductOperator(LinearOperator): ...@@ -134,7 +134,7 @@ class MatrixProductOperator(LinearOperator):
active_axes = [] active_axes = []
self._inactive_axes = list(range(len(self._domain))) self._inactive_axes = list(range(len(self._domain)))
domain_shape = [] domain_shape = []
for space_idx in spaces: for space_idx in self._spaces:
domain_shape += self._domain[space_idx].shape domain_shape += self._domain[space_idx].shape
active_axes += self._domain.axes[space_idx] active_axes += self._domain.axes[space_idx]
if space_idx in self._inactive_axes: 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