fix: fix SII implementation for NonlinearSolveNoInitCache (#506)

AayushSabharwal · ChrisRackauckas · web-flow · commit 83f8c3d43efc · 2024-11-18T13:16:49.000-01:00
Co-authored-by: Christopher Rackauckas &lt;accounts@chrisrackauckas.com&gt;
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

Original file line number	Diff line number	Diff line change
`@@ -248,6 +248,8 @@ end`
`248`	`248`	`kwargs::Any`
`249`	`249`	`end`
`250`	`250`
	`251`	`+get_u(cache::NonlinearSolveNoInitCache) = SII.state_values(cache.prob)`
	`252`	`+`
`251`	`253`	`function SciMLBase.reinit!(`
`252`	`254`	`cache::NonlinearSolveNoInitCache, u0 = cache.prob.u0; p = cache.prob.p, kwargs...`
`253`	`255`	`)`