fix

SciML · Mar 20, 2024 · cefb901 · cefb901
1 parent c5e7348
commit cefb901
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/PINO_ode_tests_gpu.jl b/test/PINO_ode_tests_gpu.jl
@@ -109,7 +109,7 @@ end
 
     opt = OptimizationOptimisers.Adam(0.001)
     alg = PINOODE(
-        flat_no, opt, train_set; init_params = ps, is_data_loss = true, is_physics_loss = true)
+        chain, opt, train_set; init_params = ps, is_data_loss = true, is_physics_loss = true)
     pino_solution = solve(prob, alg, verbose = false, maxiters = 4000)
     predict = pino_solution.predict |> cpu
     ground = u_output_