From f59f22526a01aa6e05da3405384b55cbe3a66d44 Mon Sep 17 00:00:00 2001
From: KirillZubov <kirillzubov3@gmail.com>
Date: Thu, 10 Oct 2024 15:45:56 +0400
Subject: [PATCH] update NeuralPDE

---
 src/NeuralPDE.jl |  2 +-
 test/runtests.jl | 13 -------------
 2 files changed, 1 insertion(+), 14 deletions(-)

diff --git a/src/NeuralPDE.jl b/src/NeuralPDE.jl
index 128aa72c2..7121ba3dd 100644
--- a/src/NeuralPDE.jl
+++ b/src/NeuralPDE.jl
@@ -58,7 +58,7 @@ include("BPINN_ode.jl")
 include("PDE_BPINN.jl")
 include("dgm.jl")
 
-export NNODE, NNDAE, PINOODE
+export NNODE, NNDAE, PINOODE,
        PhysicsInformedNN, discretize,
        GridTraining, StochasticTraining, QuadratureTraining, QuasiRandomTraining,
        WeightedIntervalTraining,
diff --git a/test/runtests.jl b/test/runtests.jl
index 85573f061..57e1bbe51 100644
--- a/test/runtests.jl
+++ b/test/runtests.jl
@@ -1,18 +1,5 @@
 using Pkg
 using SafeTestsets
-import NeuralPDE: NNODE, NNDAE, PINOODE
-                PhysicsInformedNN, discretize,
-                GridTraining, StochasticTraining, QuadratureTraining, QuasiRandomTraining,
-                WeightedIntervalTraining,
-                build_loss_function, get_loss_function,
-                generate_training_sets, get_variables, get_argument, get_bounds,
-                get_numeric_integral, symbolic_discretize,
-                AbstractAdaptiveLoss, NonAdaptiveLoss, GradientScaleAdaptiveLoss,
-                MiniMaxAdaptiveLoss, LogOptions,
-                ahmc_bayesian_pinn_ode, BNNODE, ahmc_bayesian_pinn_pde, vector_to_parameters,
-                BPINNsolution, BayesianPINN,
-                DeepGalerkin
-
 
 const GROUP = get(ENV, "GROUP", "All")