From d64aede32a426ca297be7adb91612488453450ff Mon Sep 17 00:00:00 2001
From: Rafael Caixeta <8386288+rmcaixeta@users.noreply.github.com>
Date: Sun, 25 Apr 2021 13:37:58 -0300
Subject: [PATCH] Clip max neighbors if n < k

---
 src/data_handling.jl | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/data_handling.jl b/src/data_handling.jl
index 1e903a7..d66a810 100644
--- a/src/data_handling.jl
+++ b/src/data_handling.jl
@@ -45,7 +45,8 @@ function get_neighbors(origin::AbstractMatrix, target::AbstractMatrix,
 	@assert nhood in [:knn,:radius] "invalid neighborhood type"
 	tree = KDTree(origin)
 	if nhood==:knn
-		idxs, dists = knn(tree, target, neigh_val, true)
+		nneigh = minimum([neigh_val, size(origin,2)])
+		idxs, dists = knn(tree, target, nneigh, true)
 		idxs, dists
 	elseif nhood==:radius
 		idxs = inrange(tree, target, neigh_val)