Revert "return None for batch"

This reverts commit 0199c312.

Revert "return None for batch"
This reverts commit 0199c312.
31638af9 · rusty1s · 0199c312 · 31638af9
Commit 31638af9 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
@@ -42,7 +42,7 @@ def grid_cluster(position, size, batch=None):
    cluster, u = consecutive(cluster)
    if batch is None:
-        return cluster, None
+        return cluster
    else:
        print(p_max.tolist(), size.tolist(), C)
        batch = (u / C).long()