fix tag, update import

SciML · Nov 19, 2024 · d6aae80 · d6aae80
1 parent 38679c5
commit d6aae80
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/lib/OrdinaryDiffEqDifferentiation/src/alg_utils.jl b/lib/OrdinaryDiffEqDifferentiation/src/alg_utils.jl
@@ -53,7 +53,7 @@ function DiffEqBase.prepare_alg(
 
     if alg_autodiff(alg) isa AutoForwardDiff && ((prob.f isa ODEFunction &&
         prob.f.f isa FunctionWrappersWrappers.FunctionWrappersWrapper) || (isbitstype(T) && sizeof(T) > 24))
-        return remake(alg, autodiff = AutoForwardDiff(chunksize = 1, tag = _get_fwd_tag(alg_autodiff(alg))))
+        return remake(alg, autodiff = AutoForwardDiff(chunksize = 1, tag = alg_autodiff(alg).tag))
     end
 
     # If the autodiff alg is AutoFiniteDiff, prob.f.f isa FunctionWrappersWrapper,

diff --git a/lib/OrdinaryDiffEqNonlinearSolve/src/OrdinaryDiffEqNonlinearSolve.jl b/lib/OrdinaryDiffEqNonlinearSolve/src/OrdinaryDiffEqNonlinearSolve.jl
@@ -25,7 +25,7 @@ import SciMLStructures: canonicalize, Tunable, isscimlstructure
 import OrdinaryDiffEqCore
 
 import SciMLOperators: islinear
-import OrdinaryDiffEqCore: nlsolve_f, set_new_W!, set_W_γdt!, _ADType_to_Bool
+import OrdinaryDiffEqCore: nlsolve_f, set_new_W!, set_W_γdt!
 
 @static if isdefined(OrdinaryDiffEqCore, :default_nlsolve)
     import OrdinaryDiffEqCore: default_nlsolve