From 2642a92df35f3b926c028ff1c0638a0d53313c3b Mon Sep 17 00:00:00 2001
From: Sathvik Bhagavan <sathvikbhagavan@gmail.com>
Date: Mon, 3 Feb 2025 17:59:49 +0530
Subject: [PATCH] fixup! test: fix ODE and PDE tests

---
 test/PINO_ode_tests.jl | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/test/PINO_ode_tests.jl b/test/PINO_ode_tests.jl
index 4f6e7e43f..08b612875 100644
--- 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]