graclus fix

Author: rusty1s

setup.py

@@ -63,7 +63,7 @@ def get_extensions():
 
 setup(
     name='torch_cluster',
-    version='1.5.0',
+    version='1.5.1',
     author='Matthias Fey',
     author_email='[email protected]',
     url='https://github.com/rusty1s/pytorch_cluster',

torch_cluster/__init__.py

@@ -3,7 +3,7 @@
 
 import torch
 
-__version__ = '1.5.0'
+__version__ = '1.5.1'
 expected_torch_version = (1, 4)
 
 try:

torch_cluster/graclus.py

@@ -33,7 +33,7 @@ def graclus_cluster(row: torch.Tensor, col: torch.Tensor,
         num_nodes = max(int(row.max()), int(col.max())) + 1
 
     # Remove self-loops.
-    mask = row == col
+    mask = row != col
     row, col = row[mask], col[mask]
 
     if weight is not None: