added pragmas

test/test_graclus.py

@@ -56,7 +56,7 @@ def test_graclus_cluster_cpu(tensor, i):
 
 @pytest.mark.skipif(not torch.cuda.is_available(), reason='no CUDA')
 @pytest.mark.parametrize('tensor,i', product(tensors, range(len(tests))))
-def test_graclus_cluster_gpu(tensor, i):
+def test_graclus_cluster_gpu(tensor, i):  # pragma: no cover
     data = tests[i]
 
     row = torch.cuda.LongTensor(data['row'])

torch_cluster/graclus.py

@@ -5,7 +5,7 @@ from .utils.ffi import graclus
 def graclus_cluster(row, col, weight=None, num_nodes=None):
     num_nodes = row.max() + 1 if num_nodes is None else num_nodes
 
-    if row.is_cuda:
+    if row.is_cuda:  # pragma: no cover
         row, col = sort_row(row, col)
     else:
         row, col = randperm(row, col)