Update geometry.py

AllenInstitute · Nov 18, 2024 · 39f4dd4 · 39f4dd4
1 parent d502cee
commit 39f4dd4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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]