fix test imports

SciML · Dec 3, 2024 · e67c595 · e67c595
1 parent 4f6e894
commit e67c595
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/test/NN_SDE_tests.jl b/test/NN_SDE_tests.jl
@@ -1,7 +1,6 @@
 @testitem "Test-1" tags=[:nnsde] begin
-    using NeuralPDE, Lux, Optimization, OptimizationOptimJL, Plots
-    using ModelingToolkit: Interval
-    using OrdinaryDiffEq, Random
+    using OrdinaryDiffEq, Random, Lux, Optimisers
+    using OptimizationOptimJL: BFGS
     Random.seed!(100)
 
     α = 1