removed rid args

torch_cluster/functions/permute.py

 import torch
 
 
-def permute(edge_index, num_nodes, rid=None, perm_edges=True):
+def random_permute(edge_index, num_nodes):
     row, col = edge_index
 
-    if perm_edges:
-        edge_rid = torch.randperm(row.size(0))
-        row, col = row[edge_rid], col[edge_rid]
+    rid = torch.randperm(row.size(0))
+    row, col = row[rid], col[rid]
 
-    rid = torch.randperm(num_nodes) if rid is None else rid
-    _, perm = rid[row].sort()
+    _, perm = rid[torch.randperm(num_nodes)].sort()
     row, col = row[perm], col[perm]
 
     return torch.stack([row, col], dim=0)

torch_cluster/functions/random.py

@@ -3,14 +3,10 @@ from .degree import node_degree
 from .permute import permute
 
 
-def random_cluster(edge_index,
-                   batch=None,
-                   rid=None,
-                   perm_edges=True,
-                   num_nodes=None):
+def random_cluster(edge_index, batch=None, num_nodes=None):
 
     num_nodes = edge_index.max() + 1 if num_nodes is None else num_nodes
-    row, col = permute(edge_index, num_nodes, rid, perm_edges)
+    row, col = permute(edge_index, num_nodes)
     degree = node_degree(row, num_nodes, out=row.new())
 
     cluster = edge_index.new(num_nodes).fill_(-1)