diff --git a/example.py b/example.py index 5210e29..e5d58db 100644 --- a/example.py +++ b/example.py @@ -14,5 +14,7 @@ def test(data): nn = kd_tree.query(X_test, 4) + radius = kd_tree.query_radius(X_test, 1.0) + if __name__ == "__main__": test(datasets)