fix: reorder system in SCCNonlinearProblem

SciML · Nov 30, 2024 · c5f6b30 · c5f6b30
1 parent 10de62b
commit c5f6b30
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/systems/nonlinear/nonlinearsystem.jl b/src/systems/nonlinear/nonlinearsystem.jl
@@ -703,6 +703,10 @@ 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
     return SCCNonlinearProblem(subprobs, explicitfuns, sys, p)
 end