From acc8fb15a3a4eef57b18c5b5de9631365fb3f647 Mon Sep 17 00:00:00 2001 From: Sam Anklesaria Date: Wed, 27 Mar 2024 09:41:59 -0500 Subject: [PATCH] Add inbounds to make_rows --- src/NearestNeighborsModule.jl | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/src/NearestNeighborsModule.jl b/src/NearestNeighborsModule.jl index 59519331..3c53a499 100644 --- a/src/NearestNeighborsModule.jl +++ b/src/NearestNeighborsModule.jl @@ -1,6 +1,5 @@ module NearestNeighborsModule using ..API -using ChainRulesCore using KernelFunctions, LinearAlgebra, SparseArrays, AbstractGPs """ @@ -21,7 +20,7 @@ function make_B(pts::AbstractVector{T}, k::Int, kern::Kernel) where {T} end function make_rows(pts::AbstractVector{T}, k::Int, kern::Kernel) where {T} - [make_row(kern, pts[max(1, i-k):i-1], pts[i]) for i in 2:length(pts)] + @inbounds [make_row(kern, pts[max(1, i-k):i-1], pts[i]) for i in 2:length(pts)] end function make_row(kern::Kernel, ns::AbstractVector{T}, p::T) where {T}