From e67c59526fac79a485be15a1d6c71ba84fa697c6 Mon Sep 17 00:00:00 2001 From: Astitva Aggarwal Date: Wed, 4 Dec 2024 03:40:52 +0530 Subject: [PATCH] fix test imports --- test/NN_SDE_tests.jl | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/test/NN_SDE_tests.jl b/test/NN_SDE_tests.jl index e410a18f6..608ee33a5 100644 --- 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