fix some depwarns

SciML · Sep 21, 2024 · 7f4a341 · 7f4a341
1 parent 6b80673
commit 7f4a341
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/test/neural_ode_mm_tests.jl b/test/neural_ode_mm_tests.jl
@@ -46,5 +46,5 @@
  optprob = Optimization.OptimizationProblem(optfunc, p)
  res = Optimization.solve(
  optprob, BFGS(; initial_stepnorm = 0.001); callback = cb, maxiters = 100)
- @test res.minimum < l1
+ @test res.objective < l1
 end
diff --git a/test/newton_neural_ode_tests.jl b/test/newton_neural_ode_tests.jl
@@ -27,7 +27,7 @@
  # KrylovTrustRegion is hardcoded to use `Array`
  psd, psax = getdata(ps), getaxes(ps)
 
- loss_function(θ) = sum(abs2, y .- stnODE(x, ComponentArray(θ, psax))[end])
+ loss_function(θ) = sum(abs2, y .- stnODE(x, ComponentArray(θ, psax)).u[end])
  l1 = loss_function(psd)
  optf = Optimization.OptimizationFunction(
  (x, p) -> loss_function(x), Optimization.AutoZygote())