Fix OrdinaryDiffEq import error

CryoGrid · Aug 7, 2024 · d7341f1 · d7341f1
1 parent d07083c
commit d7341f1
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/src/Solvers/DiffEq/DiffEq.jl b/src/Solvers/DiffEq/DiffEq.jl
@@ -38,7 +38,7 @@ function __init__()
     # OrdinaryDiffEq compatibility
     @require OrdinaryDiffEq="1dea7af3-3e70-54e6-95c3-0bf5283fa5ed" begin
         using .OrdinaryDiffEq
-        using .OrdinaryDiffEq: NLSolver
+        using .OrdinaryDiffEq: NLSolver, OrdinaryDiffEqAlgorithm, DAEAlgorithm
         # re-export selected types from OrdinaryDiffEq;
         export OrdinaryDiffEq
         # explicit methods

diff --git a/src/Solvers/DiffEq/ode_solvers.jl b/src/Solvers/DiffEq/ode_solvers.jl
@@ -1,9 +1,9 @@
 # solve/init interface
-function DiffEqBase.__solve(prob::CryoGridProblem, alg::Union{OrdinaryDiffEqAlgorithm, OrdinaryDiffEq.DAEAlgorithm}, args...; saveat=prob.saveat, kwargs...)
+function DiffEqBase.__solve(prob::CryoGridProblem, alg::Union{OrdinaryDiffEqAlgorithm, DAEAlgorithm}, args...; saveat=prob.saveat, kwargs...)
     ode_prob = ODEProblem(prob)
     return DiffEqBase.solve(ode_prob, alg, args...; saveat, kwargs...)
 end
-function DiffEqBase.__init(prob::CryoGridProblem, alg::Union{OrdinaryDiffEqAlgorithm, OrdinaryDiffEq.DAEAlgorithm}, args...; saveat=prob.saveat, kwargs...)
+function DiffEqBase.__init(prob::CryoGridProblem, alg::Union{OrdinaryDiffEqAlgorithm, DAEAlgorithm}, args...; saveat=prob.saveat, kwargs...)
     ode_prob = ODEProblem(prob)
     return DiffEqBase.init(ode_prob, alg, args...; saveat, kwargs...)
 end