fixup! refactor: correctly lower quadrature training strategy in NNODE

SciML · Mar 21, 2024 · 8f82879 · 8f82879
1 parent 2ceaa01
commit 8f82879
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/ode_solve.jl b/src/ode_solve.jl
@@ -111,7 +111,7 @@ end
 
 function generate_phi_θ(chain::Lux.AbstractExplicitLayer, t, u0, init_params)
     θ, st = Lux.setup(Random.default_rng(), chain)
-    isnothing(init_params) && init_params = θ
+    isnothing(init_params) && (init_params = θ)
     ODEPhi(chain, t, u0, st), init_params
 end