Update test/null_parameters.jl

SciML · Jun 6, 2024 · 1d9ee4a · 1d9ee4a
1 parent 1a8c408
commit 1d9ee4a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/test/null_parameters.jl b/test/null_parameters.jl
@@ -81,15 +81,15 @@ function loss10(params)
     u0 = zeros(2)
     problem = ODEProblem(dynamics, u0, (0.0, 1.0))
     rollout = solve(problem, Tsit5(), u0 = u0, p = params,
-        sensealg = GaussAdjoint(autojacvec = EnzymeVJP()))
+        sensealg = QuadratureAdjoint(autojacvec = EnzymeVJP()))
     sum(Array(rollout)[:, end])
 end
 
 function loss11(params)
     u0 = zeros(2)
     problem = ODEProblem(dynamics, u0, (0.0, 1.0))
     rollout = solve(problem, Tsit5(), u0 = u0, p = params,
-        sensealg = GaussAdjoint(autojacvec = ZygoteVJP()))
+        sensealg = QuadratureAdjoint(autojacvec = ZygoteVJP()))
     sum(Array(rollout)[:, end])
 end