fix: reorder system in SCCNonlinearProblem

SciML · Dec 1, 2024 · 03133a0 · 03133a0
1 parent 79cf520
commit 03133a0
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/systems/nonlinear/nonlinearsystem.jl b/src/systems/nonlinear/nonlinearsystem.jl
@@ -703,6 +703,11 @@ function SciMLBase.SCCNonlinearProblem{iip}(sys::NonlinearSystem, u0map,
         push!(subprobs, prob)
     end
 
+    new_dvs = dvs[reduce(vcat, var_sccs)]
+    new_eqs = eqs[reduce(vcat, eq_sccs)]
+    @set! sys.unknowns = new_dvs
+    @set! sys.eqs = new_eqs
+    sys = complete(sys)
     return SCCNonlinearProblem(subprobs, explicitfuns, sys, p)
 end