From ba63b50127247946715c4712dd266d61f7158413 Mon Sep 17 00:00:00 2001
From: Vaibhav Dixit <vaibhavyashdixit@gmail.com>
Date: Wed, 22 Nov 2023 21:43:38 -0500
Subject: [PATCH] ADd Flux test dep

---
 test/Project.toml | 2 ++
 test/minibatch.jl | 2 +-
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/test/Project.toml b/test/Project.toml
index 232624b13..a775e9da1 100644
--- 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
index 9e022d92f..1f7bd6407 100644
--- 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()