Update src/pino_ode_solve.jl

Co-authored-by: Sathvik Bhagavan <[email protected]>
SciML · Jun 12, 2024 · 0fb7d23 · 0fb7d23
1 parent fbef5c7
commit 0fb7d23
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/pino_ode_solve.jl b/src/pino_ode_solve.jl
@@ -51,6 +51,7 @@ function PINOODE(chain,
     !(chain isa Lux.AbstractExplicitLayer) && (chain = Lux.transform(chain))
     PINOODE(chain, opt, bounds, init_params, strategy, additional_loss, kwargs)
 end
+
 struct PINOPhi{C, S}
     chain::C
     st::S