return None for batch

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
+        return cluster, None
     else:
         print(p_max.tolist(), size.tolist(), C)
         batch = (u / C).long()