@@ -14,16 +14,17 @@ function generate_ODENLStepData(sys::System, u0, p, mm = calculate_massmatrix(sy
1414 end
1515 nlprob = NonlinearProblem (nlsys, op; build_initializeprob = false )
1616
17- subsetidxs = [findfirst (isequal (y),unknowns (sys)) for y in unknowns (nlsys)]
17+ subsetidxs = [findfirst (isequal (y), unknowns (sys)) for y in unknowns (nlsys)]
1818 set_gamma_c = setsym (nlsys, (ODE_GAMMA... , ODE_C))
1919 set_outer_tmp = setsym (nlsys, outer_tmp)
2020 set_inner_tmp = setsym (nlsys, inner_tmp)
2121 nlprobmap = generate_nlprobmap (sys, nlsys)
2222
23- return SciMLBase. ODENLStepData (nlprob, subsetidxs, set_gamma_c, set_outer_tmp, set_inner_tmp, nlprobmap)
23+ return SciMLBase. ODENLStepData (
24+ nlprob, subsetidxs, set_gamma_c, set_outer_tmp, set_inner_tmp, nlprobmap)
2425end
2526
26- const ODE_GAMMA = @parameters γ₁ₘₜₖ, γ₂ₘₜₖ, γ₃ₘₜₖ
27+ const ODE_GAMMA = @parameters γ₁ₘₜₖ, γ₂ₘₜₖ, γ₃ₘₜₖ
2728const ODE_C = only (@parameters cₘₜₖ)
2829
2930function get_outer_tmp (n:: Int )
@@ -55,7 +56,8 @@ function inner_nlsystem(sys::System, mm)
5556
5657 new_dvs = unknowns (sys)
5758 new_ps = [parameters (sys); [gamma1, gamma2, gamma3, c, inner_tmp, outer_tmp]]
58- nlsys = mtkcompile (System (new_eqs, new_dvs, new_ps; name = :nlsys ); split = is_split (sys))
59+ nlsys = mtkcompile (
60+ System (new_eqs, new_dvs, new_ps; name = :nlsys ); split = is_split (sys))
5961 return nlsys, outer_tmp, inner_tmp
6062end
6163
0 commit comments