From 39f4dd41a2604ae8a24e0690fe90f4126d5c2f6b Mon Sep 17 00:00:00 2001
From: Anna Grim <108307071+anna-grim@users.noreply.github.com>
Date: Mon, 18 Nov 2024 11:34:39 -0800
Subject: [PATCH] Update geometry.py

---
 src/deep_neurographs/geometry.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/deep_neurographs/geometry.py b/src/deep_neurographs/geometry.py
index b13bd16..be07134 100644
--- a/src/deep_neurographs/geometry.py
+++ b/src/deep_neurographs/geometry.py
@@ -312,7 +312,7 @@ def query_ball(kdtree, xyz, radius):
         target coordinate.
 
     """
-    idxs = kdtree.query_ball_point(xyz, radius, return_sorted=True)
+    idxs = kdtree.query_ball_point(xyz, radius)
     return kdtree.data[idxs]