From d7341f1afc8e6be516692b0cd6855d3310abe9f0 Mon Sep 17 00:00:00 2001
From: Brian Groenke <brian.groenke@awi.de>
Date: Wed, 7 Aug 2024 13:58:54 +0200
Subject: [PATCH] Fix OrdinaryDiffEq import error

---
 src/Solvers/DiffEq/DiffEq.jl      | 2 +-
 src/Solvers/DiffEq/ode_solvers.jl | 4 ++--
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/src/Solvers/DiffEq/DiffEq.jl b/src/Solvers/DiffEq/DiffEq.jl
index 9e3602c2..5066cc22 100644
--- 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
index 6b7a6668..c2aeb567 100644
--- 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