with eps

rusty1s · Jul 16, 2020 · aba2c71 · aba2c71
1 parent f445726
commit aba2c71
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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