diff --git a/Project.toml b/Project.toml index 00b6db4fb..e52bc477a 100644 --- a/Project.toml +++ b/Project.toml @@ -1,7 +1,7 @@ name = "SciMLBase" uuid = "0bca4576-84f4-4d90-8ffe-ffa030f20462" authors = ["Chris Rackauckas and contributors"] -version = "1.96.3" +version = "1.97.1" [deps] ADTypes = "47edcb42-4c32-4615-8424-f2b9edc5f35b" diff --git a/src/scimlfunctions.jl b/src/scimlfunctions.jl index ab6a8cff4..1867e1d58 100644 --- a/src/scimlfunctions.jl +++ b/src/scimlfunctions.jl @@ -1884,10 +1884,7 @@ For more details on this argument, see the ODEFunction documentation. The fields of the NonlinearFunction type directly match the names of the inputs. """ struct NonlinearFunction{iip, specialize, F, TMM, Ta, Tt, TJ, JVP, VJP, JP, SP, TW, TWt, - TPJ, - S, S2, O, TCV, - SYS, -} <: AbstractNonlinearFunction{iip} + TPJ, S, S2, O, TCV, SYS, RP} <: AbstractNonlinearFunction{iip} f::F mass_matrix::TMM analytic::Ta @@ -1905,6 +1902,7 @@ struct NonlinearFunction{iip, specialize, F, TMM, Ta, Tt, TJ, JVP, VJP, JP, SP, observed::O colorvec::TCV sys::SYS + resid_prototype::RP end TruncatedStacktraces.@truncate_stacktrace NonlinearFunction 1 2 @@ -3648,7 +3646,8 @@ function NonlinearFunction{iip, specialize}(f; DEFAULT_OBSERVED_NO_TIME, colorvec = __has_colorvec(f) ? f.colorvec : nothing, - sys = __has_sys(f) ? f.sys : nothing) where { + sys = __has_sys(f) ? f.sys : nothing, + resid_prototype = __has_resid_prototype(f) ? f.resid_prototype : nothing) where { iip, specialize, } @@ -3687,14 +3686,14 @@ function NonlinearFunction{iip, specialize}(f; Any, Any, Any, Any, Any, Any, Any, Any, Any, Any, Any, Any, typeof(syms), typeof(paramsyms), Any, - typeof(_colorvec), Any}(f, mass_matrix, + typeof(_colorvec), Any, Any}(f, mass_matrix, analytic, tgrad, jac, jvp, vjp, jac_prototype, sparsity, Wfact, Wfact_t, paramjac, syms, paramsyms, observed, - _colorvec, sys) + _colorvec, sys, resid_prototype) else NonlinearFunction{iip, specialize, typeof(f), typeof(mass_matrix), typeof(analytic), typeof(tgrad), @@ -3703,13 +3702,13 @@ function NonlinearFunction{iip, specialize}(f; typeof(Wfact_t), typeof(paramjac), typeof(syms), typeof(paramsyms), typeof(observed), - typeof(_colorvec), typeof(sys)}(f, mass_matrix, analytic, tgrad, - jac, + typeof(_colorvec), typeof(sys), typeof(resid_prototype)}(f, mass_matrix, + analytic, tgrad, jac, jvp, vjp, jac_prototype, sparsity, Wfact, Wfact_t, paramjac, syms, paramsyms, - observed, _colorvec, sys) + observed, _colorvec, sys, resid_prototype) end end @@ -3978,6 +3977,7 @@ __has_analytic(f) = isdefined(f, :analytic) __has_colorvec(f) = isdefined(f, :colorvec) __has_sys(f) = isdefined(f, :sys) __has_analytic_full(f) = isdefined(f, :analytic_full) +__has_resid_prototype(f) = isdefined(f, :resid_prototype) # compatibility has_invW(f::AbstractSciMLFunction) = false