Fix usage of AbstractVectorOfArray

SciML · Dec 11, 2024 · 86a2d26 · 86a2d26
1 parent cd9fe82
commit 86a2d26
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/lib/BoundaryValueDiffEqMIRK/src/mirk.jl b/lib/BoundaryValueDiffEqMIRK/src/mirk.jl
@@ -207,7 +207,7 @@ end
 
 # Constructing the Nonlinear Problem
 function __construct_nlproblem(
-        cache::MIRKCache{iip}, y::AbstractVector, y₀::VectorOfArray) where {iip}
+        cache::MIRKCache{iip}, y::AbstractVector, y₀::AbstractVectorOfArray) where {iip}
     pt = cache.problem_type
 
     eval_sol = EvalSol(y₀.u, cache.mesh, cache.alg, cache.k_discrete)

diff --git a/lib/BoundaryValueDiffEqMIRK/test/nlls_tests.jl b/lib/BoundaryValueDiffEqMIRK/test/nlls_tests.jl
@@ -145,8 +145,8 @@ end
     end
 
     function bc!(residual, sol, p, t)
-        y1 = first(sol)
-        y2 = last(sol)
+        y1 = sol(0.0)
+        y2 = sol(0.5)
         R0_u = reshape(@view(y1[4:12]), 3, 3)
         RL_u = reshape(@view(y2[4:12]), 3, 3)