diff --git a/lib/NonlinearSolveBase/src/abstract_types.jl b/lib/NonlinearSolveBase/src/abstract_types.jl index 6105f5125..6829e19c3 100644 --- a/lib/NonlinearSolveBase/src/abstract_types.jl +++ b/lib/NonlinearSolveBase/src/abstract_types.jl @@ -307,7 +307,7 @@ SciMLBase.isinplace(cache::AbstractNonlinearSolveCache) = SciMLBase.isinplace(ca ## SII Interface SII.symbolic_container(cache::AbstractNonlinearSolveCache) = cache.prob SII.parameter_values(cache::AbstractNonlinearSolveCache) = SII.parameter_values(cache.prob) -SII.state_values(cache::AbstractNonlinearSolveCache) = SII.state_values(cache.prob) +SII.state_values(cache::AbstractNonlinearSolveCache) = get_u(cache) function Base.getproperty(cache::AbstractNonlinearSolveCache, sym::Symbol) if sym === :ps diff --git a/lib/NonlinearSolveBase/src/polyalg.jl b/lib/NonlinearSolveBase/src/polyalg.jl index 54b61998f..76ba500dd 100644 --- a/lib/NonlinearSolveBase/src/polyalg.jl +++ b/lib/NonlinearSolveBase/src/polyalg.jl @@ -64,7 +64,7 @@ end function SII.symbolic_container(cache::NonlinearSolvePolyAlgorithmCache) return cache.caches[cache.current] end -SII.state_values(cache::NonlinearSolvePolyAlgorithmCache) = cache.u0 +SII.state_values(cache::NonlinearSolvePolyAlgorithmCache) = SII.state_values(SII.symbolic_container(cache)) function Base.show(io::IO, ::MIME"text/plain", cache::NonlinearSolvePolyAlgorithmCache) println(io, "NonlinearSolvePolyAlgorithmCache with \