From 0cbbae65ec40702f8c3262a5af9515d509c15d0e Mon Sep 17 00:00:00 2001
From: Aayush Sabharwal <aayush.sabharwal@gmail.com>
Date: Thu, 12 Dec 2024 14:36:49 +0530
Subject: [PATCH] fix: remove type piracy of `alg_autodiff`

---
 src/alg_utils.jl | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/alg_utils.jl b/src/alg_utils.jl
index 4d22b89e..ee4b86a5 100644
--- 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()