Skip to content

Commit 3b9bca9

Browse files
Merge pull request #601 from jClugstor/test_fixes
Fix complex test, add _alg_autodiff method for Composite algs
2 parents e8956b9 + c3441c8 commit 3b9bca9

File tree

2 files changed

+2
-1
lines changed

2 files changed

+2
-1
lines changed

src/alg_utils.jl

+1
Original file line numberDiff line numberDiff line change
@@ -23,6 +23,7 @@ _alg_autodiff(::StochasticDiffEqNewtonAlgorithm{T, AD}) where {T, AD} = Val{AD}(
2323
_alg_autodiff(::StochasticDiffEqNewtonAdaptiveAlgorithm{T, AD}) where {T, AD} = Val{AD}()
2424
_alg_autodiff(::StochasticDiffEqJumpNewtonAdaptiveAlgorithm{T, AD}) where {T, AD} = Val{AD}()
2525
_alg_autodiff(::StochasticDiffEqJumpNewtonDiffusionAdaptiveAlgorithm{T, AD}) where {T, AD} = Val{AD}()
26+
_alg_autodiff(alg::StochasticCompositeAlgorithm) = _alg_autodiff(alg.algs[end])
2627

2728
function OrdinaryDiffEqCore.alg_autodiff(alg::Union{StochasticDiffEqAlgorithm, StochasticDiffEqRODEAlgorithm})
2829
ad = _alg_autodiff(alg)

test/complex_tests.jl

+1-1
Original file line numberDiff line numberDiff line change
@@ -39,6 +39,6 @@ end
3939

4040
# currently broken
4141
for alg in implicit_autodiff
42-
@test_throws ArgumentError solve(prob, alg)
42+
@test_throws StochasticDiffEq.OrdinaryDiffEqDifferentiation.FirstAutodiffJacError solve(prob, alg)
4343
end
4444
end

0 commit comments

Comments
 (0)