fix(LinearInterpolation): better check for arrays

SciML · Sep 23, 2024 · 9530aa6 · 9530aa6
1 parent 2ffef8d
commit 9530aa6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/parameter_caches.jl b/src/parameter_caches.jl
@@ -18,8 +18,8 @@ function safe_diff(b, a::T) where {T}
     b == a ? zero(T) : b - a
 end
 
-function linear_interpolation_parameters(u::AbstractArray{T}, t, idx) where {T}
-    Δu = if u isa AbstractArray 
+function linear_interpolation_parameters(u::AbstractArray{T, N}, t, idx) where {T, N}
+    Δu = if N > 1
         ax = axes(u)
         safe_diff.(u[ax[1:end-1]..., idx+1:idx+1] , u[ax[1:end-1]..., idx:idx])
     else