fix(SciML): Make 3D compatible

agdestein · Jul 15, 2024 · 5bcd07e · 5bcd07e
1 parent c640c2f
commit 5bcd07e
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/lib/SciMLCompat/main.jl b/lib/SciMLCompat/main.jl
@@ -18,7 +18,7 @@ f = create_right_hand_side(setup, psolver)
 f(stack(ustart), nothing, 0.0)
 
 # Solve the ODE using SciML
-prob = ODEProblem(f, stack(ustart), (0.0, 1))
+prob = ODEProblem(f, stack(ustart), (0.0, 1.0))
 sol = solve(
     prob,
     Tsit5();

diff --git a/lib/SciMLCompat/src/SciMLCompat.jl b/lib/SciMLCompat/src/SciMLCompat.jl
@@ -10,7 +10,7 @@ the OrdinaryDiffEq ODE solvers. Note that `u` has to be an array.
 To convert the tuple `u = (ux, uy)` to an array, use `stack(u)`.
 """
 create_right_hand_side(setup, psolver) = function right_hand_side(u, p, t)
-    u = eachslice(u; dims = 3)
+    u = eachslice(u; dims = ndims(u))
     u = (u...,)
     u = INS.apply_bc_u(u, t, setup)
     F = INS.momentum(u, nothing, t, setup)