Skip to content

Commit d51f20e

Browse files
committed
add missing checkbounds kwarg to get_torn_eqs_vars
1 parent 6dbe08b commit d51f20e

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

src/structural_transformation/codegen.jl

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -166,7 +166,7 @@ function gen_nlsolve(sys, eqs, vars; checkbounds=true)
166166
]
167167
end
168168

169-
function get_torn_eqs_vars(sys)
169+
function get_torn_eqs_vars(sys; checkbounds=true)
170170
s = structure(sys)
171171
partitions = s.partitions
172172
vars = s.fullvars
@@ -175,7 +175,7 @@ function get_torn_eqs_vars(sys)
175175
torn_eqs = map(idxs-> eqs[idxs], map(x->x.e_residual, partitions))
176176
torn_vars = map(idxs->vars[idxs], map(x->x.v_residual, partitions))
177177

178-
gen_nlsolve.((sys,), torn_eqs, torn_vars)
178+
gen_nlsolve.((sys,), torn_eqs, torn_vars, checkbounds=checkbounds)
179179
end
180180

181181
function build_torn_function(
@@ -212,7 +212,7 @@ function build_torn_function(
212212
],
213213
[],
214214
Let(
215-
collect(Iterators.flatten(get_torn_eqs_vars(sys))),
215+
collect(Iterators.flatten(get_torn_eqs_vars(sys, checkbounds=checkbounds))),
216216
odefunbody
217217
)
218218
)

0 commit comments

Comments
 (0)