skip metis test

fcf15650 · rusty1s · 80fe5c0f · fcf15650
Commit fcf15650 authored Mar 11, 2022 by rusty1s
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 2 deletions

test/test_metis.py test/test_metis.py +4 -2

No files found.
--- a/test/test_metis.py
+++ b/test/test_metis.py
-import pytest
 from itertools import product

+import pytest
 import torch
 from torch_sparse.tensor import SparseTensor

 from .utils import devices

 try:
-    torch.ops.torch_sparse.partition
+    rowptr = torch.tensor([0, 1])
+    col = torch.tensor([0])
+    torch.ops.torch_sparse.partition(rowptr, col, None, 1)
    with_metis = True
 except RuntimeError:
    with_metis = False