typo

bea734f1 · rusty1s · 548201c9 · bea734f1
Commit bea734f1 authored Nov 19, 2018 by rusty1s
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 4 deletions

torch_cluster/radius.py torch_cluster/radius.py +2 -4

No files found.
--- a/torch_cluster/radius.py
+++ b/torch_cluster/radius.py
@@ -72,10 +72,8 @@ def radius_graph(x, r, batch=None, max_num_neighbors=32):
    Examples::
        >>> x = torch.Tensor([[-1, -1], [-1, 1], [1, -1], [1, 1]])
-        >>> batch_x = torch.Tensor([0, 0, 0, 0])
+        >>> batch = torch.Tensor([0, 0, 0, 0])
-        >>> y = torch.Tensor([[-1, 0], [1, 0]])
+        >>> out = radius_graph(x, 1.5, batch)
-        >>> batch_x = torch.Tensor([0, 0])
-        >>> out = radius(x, y, 1.5, batch_x, batch_y)
    """
    edge_index = radius(x, x, r, batch, batch, max_num_neighbors + 1)