Skip to content

Commit 4fc319b

Browse files
committed
Fix #1478
1 parent a965fa2 commit 4fc319b

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

src/structural_transformation/codegen.jl

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -433,6 +433,11 @@ function build_observed_function(
433433
union!(required_algvars, intersect(algvars, vs))
434434
empty!(vs)
435435
end
436+
for eq in assignments
437+
vars!(vs, eq.rhs)
438+
union!(required_algvars, intersect(algvars, vs))
439+
empty!(vs)
440+
end
436441

437442
varidxs = findall(x->x in required_algvars, fullvars)
438443
subset = find_solve_sequence(var_sccs, varidxs)
@@ -473,8 +478,8 @@ function build_observed_function(
473478
[],
474479
pre(Let(
475480
[
476-
assignments[is_not_prepended_assignment]
477481
collect(Iterators.flatten(solves))
482+
assignments[is_not_prepended_assignment]
478483
map(eq -> eq.lhseq.rhs, obs[1:maxidx])
479484
subs
480485
],

0 commit comments

Comments
 (0)