with eps

aba2c714 · rusty1s · f445726c · aba2c714
Commit aba2c714 authored Jul 16, 2020 by rusty1s
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

test/test_radius.py test/test_radius.py +1 -1

No files found.
--- a/test/test_radius.py
+++ b/test/test_radius.py
@@ -67,7 +67,7 @@ def test_radius_graph_large(dtype, device):
                              max_num_neighbors=2000, num_workers=6)

    tree = scipy.spatial.cKDTree(x.numpy())
-    col = tree.query_ball_point(x.cpu(), r=0.5)
+    col = tree.query_ball_point(x.cpu(), r=0.5 + 0.00001)
    truth = set([(i, j) for i, ns in enumerate(col) for j in ns])

    assert to_set(edge_index) == truth