typo

Author: rusty1s

test/test_graclus.py

@@ -41,7 +41,7 @@ def assert_correct_graclus(row, col, cluster):
 
 
 @pytest.mark.parametrize('test,dtype,device', product(tests, dtypes, devices))
-def test_graclus_cluster_cpu(test, dtype, device):
+def test_graclus_cluster(test, dtype, device):
     row = tensor(test['row'], torch.long, device)
     col = tensor(test['col'], torch.long, device)
     weight = tensor(test.get('weight'), dtype, device)

test/test_grid.py

@@ -27,7 +27,7 @@
 
 
 @pytest.mark.parametrize('test,dtype,device', product(tests, dtypes, devices))
-def test_grid_cluster_cpu(test, dtype, device):
+def test_grid_cluster(test, dtype, device):
     pos = tensor(test['pos'], dtype, device)
     size = tensor(test['size'], dtype, device)
     start = tensor(test.get('start'), dtype, device)

test/utils.py

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