diff --git a/test/test_rw.py b/test/test_rw.py index 27bebc4..6afbf61 100644 --- a/test/test_rw.py +++ b/test/test_rw.py @@ -84,7 +84,6 @@ def test_rw_small_with_edge_indices(device): ] - @pytest.mark.parametrize('device', devices) def test_rw_large_with_edge_weights(device): row = tensor([0, 1, 1, 1, 2, 2, 3, 3, 4, 4], torch.long, device) @@ -123,4 +122,4 @@ def test_rw_small_with_edge_weights(device): [0, 1, 0, 1, 0], [1, 0, 1, 0, 1], [2, 2, 2, 2, 2], - ] \ No newline at end of file + ]