Fix has_ for DynamicalBVPFunction

Signed-off-by: ErikQQY <[email protected]>
SciML · Apr 15, 2024 · 139522d · 139522d
1 parent 31ff6a1
commit 139522d
Showing 1 changed file with 17 additions and 8 deletions.
diff --git a/src/scimlfunctions.jl b/src/scimlfunctions.jl
@@ -4279,14 +4279,23 @@ has_Wfact_t(f::Union{SplitFunction, SplitSDEFunction}) = has_Wfact_t(f.f1)
 has_paramjac(f::Union{SplitFunction, SplitSDEFunction}) = has_paramjac(f.f1)
 has_colorvec(f::Union{SplitFunction, SplitSDEFunction}) = has_colorvec(f.f1)
 
-has_jac(f::Union{DynamicalODEFunction, DynamicalDDEFunction, DynamicalBVPFunction}) = has_jac(f.f1)
-has_jvp(f::Union{DynamicalODEFunction, DynamicalDDEFunction, DynamicalBVPFunction}) = has_jvp(f.f1)
-has_vjp(f::Union{DynamicalODEFunction, DynamicalDDEFunction, DynamicalBVPFunction}) = has_vjp(f.f1)
-has_tgrad(f::Union{DynamicalODEFunction, DynamicalDDEFunction, DynamicalBVPFunction}) = has_tgrad(f.f1)
-has_Wfact(f::Union{DynamicalODEFunction, DynamicalDDEFunction, DynamicalBVPFunction}) = has_Wfact(f.f1)
-has_Wfact_t(f::Union{DynamicalODEFunction, DynamicalDDEFunction, DynamicalBVPFunction}) = has_Wfact_t(f.f1)
-has_paramjac(f::Union{DynamicalODEFunction, DynamicalDDEFunction, DynamicalBVPFunction}) = has_paramjac(f.f1)
-has_colorvec(f::Union{DynamicalODEFunction, DynamicalDDEFunction, DynamicalBVPFunction}) = has_colorvec(f.f1)
+has_jac(f::Union{DynamicalODEFunction, DynamicalDDEFunction}) = has_jac(f.f1)
+has_jvp(f::Union{DynamicalODEFunction, DynamicalDDEFunction}) = has_jvp(f.f1)
+has_vjp(f::Union{DynamicalODEFunction, DynamicalDDEFunction}) = has_vjp(f.f1)
+has_tgrad(f::Union{DynamicalODEFunction, DynamicalDDEFunction}) = has_tgrad(f.f1)
+has_Wfact(f::Union{DynamicalODEFunction, DynamicalDDEFunction}) = has_Wfact(f.f1)
+has_Wfact_t(f::Union{DynamicalODEFunction, DynamicalDDEFunction}) = has_Wfact_t(f.f1)
+has_paramjac(f::Union{DynamicalODEFunction, DynamicalDDEFunction}) = has_paramjac(f.f1)
+has_colorvec(f::Union{DynamicalODEFunction, DynamicalDDEFunction}) = has_colorvec(f.f1)
+
+has_jac(f::DynamicalBVPFunction) = has_jac(f.f)
+has_jvp(f::DynamicalBVPFunction) = has_jvp(f.f)
+has_vjp(f::DynamicalBVPFunction) = has_vjp(f.f)
+has_tgrad(f::DynamicalBVPFunction) = has_tgrad(f.f)
+has_Wfact(f::DynamicalBVPFunction) = has_Wfact(f.f)
+has_Wfact_t(f::DynamicalBVPFunction) = has_Wfact_t(f.f)
+has_paramjac(f::DynamicalBVPFunction) = has_paramjac(f.f)
+has_colorvec(f::DynamicalBVPFunction) = has_colorvec(f.f)
 
 has_jac(f::Union{UDerivativeWrapper, UJacobianWrapper}) = has_jac(f.f)
 has_jvp(f::Union{UDerivativeWrapper, UJacobianWrapper}) = has_jvp(f.f)