argument order

fed36b1f · rusty1s · cd250a63 · fed36b1f
Commit fed36b1f authored Feb 19, 2018 by rusty1s
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

torch_cluster/functions/grid.py torch_cluster/functions/grid.py +1 -1

No files found.
--- a/torch_cluster/functions/grid.py
+++ b/torch_cluster/functions/grid.py
@@ -3,7 +3,7 @@ import torch
 from .utils import get_func, consecutive


-def grid_cluster(position, size, batch=None, fake_nodes=False, offset=None):
+def grid_cluster(position, size, batch=None, offset=None, fake_nodes=False):
    # Allow one-dimensional positions.
    if position.dim() == 1:
        position = position.unsqueeze(-1)