update NeuralODE

ChrisRackauckas · ChrisRackauckas · commit e737613e70e5 · 2023-04-20T05:45:22.000+01:00
diff --git a/test/diffeqfluxtests.jl b/test/diffeqfluxtests.jl
@@ -77,7 +77,7 @@ pp, st = Lux.setup(rng, dudt2)
 pp = ComponentArray(pp)
 
 function predict_neuralode(p)
-    Array(prob_neuralode(u0, p))
+    Array(prob_neuralode(u0, p, st)[1])
 end
 
 function loss_neuralode(p)
