diff options
| -rw-r--r-- | polymatrix/expression/utils/getvariableindices.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/polymatrix/expression/utils/getvariableindices.py b/polymatrix/expression/utils/getvariableindices.py index 7010d23..08b44d3 100644 --- a/polymatrix/expression/utils/getvariableindices.py +++ b/polymatrix/expression/utils/getvariableindices.py @@ -33,6 +33,6 @@ def get_variable_indices(state, variables): yield variable else: - yield state.offset_dict[variable][0] + yield global_state[0].offset_dict[variable][0] - return state, tuple(gen_indices()) + return global_state[0], tuple(gen_indices()) |