ADd Flux test dep

SciML · Nov 23, 2023 · ba63b50 · ba63b50
1 parent 07e77d1
commit ba63b50
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/test/Project.toml b/test/Project.toml
@@ -5,6 +5,7 @@ ComponentArrays = "b0b7db55-cfe3-40fc-9ded-d10e2dbeff66"
 DiffEqFlux = "aae7a2af-3d4f-5e19-a356-7da93b79d9d0"
 Enzyme = "7da242da-08ed-463a-9acd-ee780be4f1d9"
 FiniteDiff = "6a86dc24-6348-571c-b903-95158fe2bd41"
+Flux = "587475ba-b771-5e3f-ad9e-33799f191a9c"
 ForwardDiff = "f6369f11-7733-5829-9624-2563aa707210"
 IterTools = "c8e1da08-722c-5040-9ed9-7db0dc04731e"
 Lux = "b2108857-7c20-44ae-9111-449ecde12c47"
@@ -29,6 +30,7 @@ ComponentArrays = ">= 0.13.9"
 DiffEqFlux = ">= 2"
 Enzyme = ">= 0.11.0"
 FiniteDiff = ">= 2.8.1"
+Flux = "0.14"
 ForwardDiff = ">= 0.10.19"
 IterTools = ">= 1.3.0"
 Lux = ">= 0.4.50"

diff --git a/test/minibatch.jl b/test/minibatch.jl
@@ -1,5 +1,5 @@
 using DiffEqFlux, Optimization, OrdinaryDiffEq, OptimizationOptimisers, ModelingToolkit,
-    SciMLSensitivity, Lux, Random, ComponentArrays
+    SciMLSensitivity, Lux, Random, ComponentArrays, Flux
 
 rng = Random.default_rng()