fix: fix SII implementation for NonlinearSolveNoInitCache (#506)

Co-authored-by: Christopher Rackauckas <[email protected]>
SciML · Nov 18, 2024 · 83f8c3d · 83f8c3d
1 parent 7f90cf4
commit 83f8c3d
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/lib/NonlinearSolveBase/src/solve.jl b/lib/NonlinearSolveBase/src/solve.jl
@@ -248,6 +248,8 @@ end
     kwargs::Any
 end
 
+get_u(cache::NonlinearSolveNoInitCache) = SII.state_values(cache.prob)
+
 function SciMLBase.reinit!(
         cache::NonlinearSolveNoInitCache, u0 = cache.prob.u0; p = cache.prob.p, kwargs...
 )

diff --git a/test/mtk_cache_indexing_tests.jl b/test/mtk_cache_indexing_tests.jl
@@ -20,6 +20,9 @@
         nint = init(nlprob, alg)
         @test nint isa integtype
 
+        @test_nowarn state_values(nint)
+        @test_nowarn parameter_values(nint)
+
         for (i, sym) in enumerate([X, nlsys.X, :X])
             # test both getindex and setindex!
             nint[sym] = 1.5i