fix: fix SII implementation for NonlinearSolveNoInitCache

SciML · Nov 18, 2024 · 73e381f · 73e381f
1 parent 43e6b3d
commit 73e381f
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