Small fixes

SciML · Jul 9, 2024 · 436ac01 · 436ac01
1 parent aa23c78
commit 436ac01
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/ext/DataInterpolationsChainRulesCoreExt.jl b/ext/DataInterpolationsChainRulesCoreExt.jl
@@ -70,7 +70,7 @@ end
 function ChainRulesCore.rrule(::typeof(interpolation_parameters),
         ::Val{:QuadraticSpline},
         z::AbstractVector, t::AbstractVector, idx::Integer)
-    σ = interpolation_parameters(Val(:QuadraticSpline), z, t, idx)s
+    σ = interpolation_parameters(Val(:QuadraticSpline), z, t, idx)
     dz = SparseVector(length(z), [idx, idx + 1], zeros(2))
     Δt = t[idx + 1] - t[idx]
 

diff --git a/src/interpolation_caches.jl b/src/interpolation_caches.jl
@@ -299,7 +299,7 @@ function QuadraticSpline(
     # zero for element type of d, which we don't know yet
     typed_zero = zero(2 // 1 * (u[begin + 1] - u[begin]) / (t[begin + 1] - t[begin]))
 
-    d = map(i -> i == 1 ? typed_zero : 2 // 1 * (u[i] - u[i - 1]) / (t[i] - t[i - 1]), 1:s)
+    d = [2 // 1 * (u[i] - u[max(1, i - 1)]) / (t[i] - t[1 + abs(i - 2)]) for i in eachindex(t)]
     z = tA \ d
     p = QuadraticSplineParameterCache(z, t)
     A = QuadraticSpline(u, t, nothing, p, tA, d, z, extrapolate, safetycopy)