Skip to content

Commit 9b69e0d

Browse files
fix: handle scalarized array symbolics in distribute_shift
1 parent 44f71a8 commit 9b69e0d

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/structural_transformation/utils.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -596,7 +596,7 @@ function _distribute_shift(expr, shift)
596596
(op isa Pre || op isa Initial) && return expr
597597
args = arguments(expr)
598598

599-
if ModelingToolkit.isvariable(expr)
599+
if ModelingToolkit.isvariable(expr) && operation(expr) !== getindex
600600
(length(args) == 1 && isequal(shift.t, only(args))) ? (return shift(expr)) :
601601
(return expr)
602602
elseif op isa Shift

0 commit comments

Comments
 (0)