Removed general dependencies in favor of specialized packages

Project.toml

@@ -7,7 +7,6 @@ version = "0.4.0"
 CUDA = "052768ef-5323-5732-b1bb-66c8b64840ba"
 ChainRulesCore = "d360d2e6-b24c-11e9-a2a3-2a2ae2dbcce4"
 DataFrames = "a93c6f00-e57d-5684-b7b6-d8193f3e46c0"
-DifferentialEquations = "0c46a032-eb83-5123-abaf-570d42b7fbaa"
 Distributions = "31c24e10-a181-5473-b8eb-7969acd0382f"
 GraphNetCore = "7809f980-de1b-4f9a-8451-85f041491431"
 HDF5 = "f67ccb44-e63f-5c2f-98bd-6dc0ccc4ba2f"
@@ -16,6 +15,7 @@ JSON = "682c06a0-de6a-54ab-a142-c8b1cf79cde6"
 Lux = "b2108857-7c20-44ae-9111-449ecde12c47"
 LuxCUDA = "d0bbae9a-e099-4d5b-a835-1c6931763bda"
 Optimisers = "3bd65402-5787-11e9-1adc-39752487f4e2"
+OrdinaryDiffEq = "1dea7af3-3e70-54e6-95c3-0bf5283fa5ed"
 Printf = "de0858da-6303-5e67-8744-51eddeeeb8d7"
 ProgressMeter = "92933f4c-e287-5a05-a399-4b506db050ca"
 Random = "9a3f8284-a2c9-5f02-9a11-845980a1fd5c"
@@ -32,7 +32,6 @@ Aqua = "0.8"
 CUDA = "5"
 ChainRulesCore = "1.16.0 - 1"
 DataFrames = "1.6"
-DifferentialEquations = "7.11 - 7"
 Distributions = "0.25"
 GraphNetCore = "0.3"
 HDF5 = "0.17"
@@ -41,6 +40,7 @@ JSON = "0.21"
 Lux = "0.5"
 LuxCUDA = "0.3"
 Optimisers = "0.3"
+OrdinaryDiffEq = "6.85 - 6"
 Printf = "1"
 ProgressMeter = "1.7.0 - 1"
 Random = "1"

examples/cylinder_flow/cylinder_flow.jl

@@ -5,7 +5,7 @@
 
 using MeshGraphNets
 
-import DifferentialEquations: Euler, Tsit5
+import OrdinaryDiffEq: Euler, Tsit5
 import Optimisers: Adam
 
 ######################

src/MeshGraphNets.jl

@@ -13,14 +13,14 @@ using Optimisers
 using Wandb
 using Zygote
 
-import DifferentialEquations.OrdinaryDiffEq: OrdinaryDiffEqAlgorithm, Tsit5
+import OrdinaryDiffEq: OrdinaryDiffEqAlgorithm, Tsit5
 import ProgressMeter: Progress
 import SciMLBase: ODEProblem
 
 import Base: @kwdef
-import DifferentialEquations: solve, remake
 import HDF5: h5open, create_group, open_group
 import ProgressMeter: next!, update!, finish!
+import SciMLBase: solve, remake
 import Statistics: mean
 
 include("utils.jl")

src/dataset.jl

@@ -4,10 +4,11 @@
 #
 
 import Distributions: Normal
+import HDF5: Group
 import Random: MersenneTwister
 import TFRecord: Example
 
-import HDF5: h5open, Group, read_dataset
+import HDF5: read_dataset
 import JLD2: jldopen
 import JSON: parse
 import Random: seed!, make_seed, shuffle

src/strategies.jl

@@ -3,8 +3,7 @@
 # Licensed under the MIT license. See LICENSE file in the project root for details.
 #
 
-import DifferentialEquations: ODEFunction
-import SciMLBase: AbstractSensitivityAlgorithm
+import SciMLBase: AbstractSensitivityAlgorithm, ODEFunction
 import SciMLSensitivity: InterpolatingAdjoint, ZygoteVJP
 
 #######################################################

src/utils.jl

@@ -4,7 +4,7 @@
 #
 
 import Printf: @sprintf
-import Statistics: mean, stdm
+import Statistics: stdm
 
 """
     der_minmax(path)