Merge pull request #595 from AayushSabharwal/as/fix-trait

fix: remove type piracy of `alg_autodiff`
SciML · Dec 12, 2024 · 75c7627 · 75c7627
2 parents 7ea7f8e + 0cbbae6
commit 75c7627
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/alg_utils.jl b/src/alg_utils.jl
@@ -24,7 +24,7 @@ _alg_autodiff(::StochasticDiffEqNewtonAdaptiveAlgorithm{T, AD}) where {T, AD} =
 _alg_autodiff(::StochasticDiffEqJumpNewtonAdaptiveAlgorithm{T, AD}) where {T, AD} = Val{AD}()
 _alg_autodiff(::StochasticDiffEqJumpNewtonDiffusionAdaptiveAlgorithm{T, AD}) where {T, AD} = Val{AD}()
 
-function OrdinaryDiffEqCore.alg_autodiff(alg)
+function OrdinaryDiffEqCore.alg_autodiff(alg::Union{StochasticDiffEqAlgorithm, StochasticDiffEqRODEAlgorithm})
     ad = _alg_autodiff(alg)
     if ad == Val(false)
         return ADTypes.AutoFiniteDiff()