fix: fix SII implementation for caches

Author: AayushSabharwal

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

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 \