fix

SciML · Jul 12, 2024 · 61fa804 · 61fa804
1 parent 10e08c2
commit 61fa804
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/test/PINO_ode_tests.jl b/test/PINO_ode_tests.jl
@@ -2,7 +2,7 @@ using Test
 using OptimizationOptimisers
 using Lux
 using Statistics, Random
-using LuxNeuralOperators
+using NeuralOperators
 using NeuralPDE
 
 # dG(u(t, p), t) = f(G,u(t, p))
@@ -16,6 +16,9 @@ using NeuralPDE
             Dense(1 => 10, Lux.tanh_fast), Dense(10 => 10, Lux.tanh_fast), Dense(10 => 10)),
         Chain(Dense(1 => 10, Lux.tanh_fast), Dense(10 => 10, Lux.tanh_fast),
             Dense(10 => 10, Lux.tanh_fast)))
+    #TODO test FNO
+            fno = FourierNeuralOperator(gelu; chs = (2, 64, 64, 128, 1), modes = (16,))
+
     u = rand(1, 50)
     v = rand(1, 40, 1)
     θ, st = Lux.setup(Random.default_rng(), deeponet)