coveragerc

.coveragerc

+[report]
+exclude_lines =
+    pragma: no cover
+    cuda

test/utils.py

@@ -5,7 +5,7 @@ dtypes = get_all_dtypes()
 dtypes.remove(torch.half)
 
 devices = [torch.device('cpu')]
-if torch.cuda.is_available():  # pragma: no cover
+if torch.cuda.is_available():
     devices += [torch.device('cuda:{}'.format(torch.cuda.current_device()))]
 
 

torch_cluster/graclus.py

@@ -23,7 +23,7 @@ def graclus_cluster(row, col, weight=None, num_nodes=None):
 
     num_nodes = row.max().item() + 1 if num_nodes is None else num_nodes
 
-    if row.is_cuda:  # pragma: no cover
+    if row.is_cuda:
         row, col = sort_row(row, col)
     else:
         row, col = randperm(row, col)