a few fixes

SciML · Jun 6, 2024 · 1a8c408 · 1a8c408
1 parent 89d38f7
commit 1a8c408
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/src/gauss_adjoint.jl b/src/gauss_adjoint.jl
@@ -424,7 +424,6 @@ function GaussIntegrand(sol, sensealg, checkpoints, dgdp = nothing)
     else
         pf = DiffEqBase.ParamJacobianWrapper(unwrappedf, tspan[1], y)
         pJ = similar(u0, length(u0), numparams)
-        pJ .= 0
         paramjac_config = build_param_jac_config(sensealg, pf, y, p)
     end
 

diff --git a/test/sde_nondiag_stratonovich.jl b/test/sde_nondiag_stratonovich.jl
@@ -677,7 +677,7 @@ end
     _dp1 = compute_dp(p, prob!, ForwardDiffSensitivity())
     _dp2 = compute_dp(p, prob!, BacksolveAdjoint(autojacvec = ReverseDiffVJP()))
     _dp3 = compute_dp(p, prob!, InterpolatingAdjoint(autojacvec = ReverseDiffVJP()))
-    @test_broken !any(isnan, compute_dp(p, prob!, InterpolatingAdjoint()))
+    @test !any(isnan, compute_dp(p, prob!, InterpolatingAdjoint()))
 
     @test dp1≈_dp1 rtol=1e-8
     @test dp2≈_dp2 rtol=1e-8