fixup! test: fix ODE and PDE tests

SciML · Feb 3, 2025 · 2642a92 · 2642a92
1 parent 5788f24
commit 2642a92
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/PINO_ode_tests.jl b/test/PINO_ode_tests.jl
@@ -231,7 +231,7 @@ end
         Chain(Dense(1 => 10, Lux.tanh_fast), Dense(10 => 10, Lux.tanh_fast),
             Dense(10 => 10, Lux.tanh_fast)))
 
-    u = rand(2, 50)
+    u = rand(3, 50)
     v = rand(1, 40, 1)
     θ, st = Lux.setup(Random.default_rng(), deeponet)
     c = deeponet((u, v), θ, st)[1]