diff --git a/src/systems/solver_nlprob.jl b/src/systems/solver_nlprob.jl index 463f25e5e6..9fdc85eacd 100644 --- a/src/systems/solver_nlprob.jl +++ b/src/systems/solver_nlprob.jl @@ -14,16 +14,17 @@ function generate_ODENLStepData(sys::System, u0, p, mm = calculate_massmatrix(sy end nlprob = NonlinearProblem(nlsys, op; build_initializeprob = false) - subsetidxs = [findfirst(isequal(y),unknowns(sys)) for y in unknowns(nlsys)] + subsetidxs = [findfirst(isequal(y), unknowns(sys)) for y in unknowns(nlsys)] set_gamma_c = setsym(nlsys, (ODE_GAMMA..., ODE_C)) set_outer_tmp = setsym(nlsys, outer_tmp) set_inner_tmp = setsym(nlsys, inner_tmp) nlprobmap = generate_nlprobmap(sys, nlsys) - return SciMLBase.ODENLStepData(nlprob, subsetidxs, set_gamma_c, set_outer_tmp, set_inner_tmp, nlprobmap) + return SciMLBase.ODENLStepData( + nlprob, subsetidxs, set_gamma_c, set_outer_tmp, set_inner_tmp, nlprobmap) end -const ODE_GAMMA = @parameters γ₁ₘₜₖ, γ₂ₘₜₖ, γ₃ₘₜₖ +const ODE_GAMMA = @parameters γ₁ₘₜₖ, γ₂ₘₜₖ, γ₃ₘₜₖ const ODE_C = only(@parameters cₘₜₖ) function get_outer_tmp(n::Int) @@ -55,7 +56,8 @@ function inner_nlsystem(sys::System, mm) new_dvs = unknowns(sys) new_ps = [parameters(sys); [gamma1, gamma2, gamma3, c, inner_tmp, outer_tmp]] - nlsys = mtkcompile(System(new_eqs, new_dvs, new_ps; name = :nlsys); split = is_split(sys)) + nlsys = mtkcompile( + System(new_eqs, new_dvs, new_ps; name = :nlsys); split = is_split(sys)) return nlsys, outer_tmp, inner_tmp end