fix: reorder system in SCCNonlinearProblem

SciML · Nov 26, 2024 · 38c2906 · 38c2906
1 parent 5910eb5
commit 38c2906
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/systems/nonlinear/nonlinearsystem.jl b/src/systems/nonlinear/nonlinearsystem.jl
@@ -698,6 +698,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