update

csrc/metis.cpp

-#include "cpu/metis_cpu.h"
-
 #include <Python.h>
 #include <torch/script.h>
 
+#include "cpu/metis_cpu.h"
+
 #ifdef _WIN32
 PyMODINIT_FUNC PyInit__metis(void) { return NULL; }
 #endif

test/test_metis.py

@@ -12,11 +12,12 @@ def test_metis(device):
     assert partptr.numel() == 3
     assert perm.numel() == 6
 
-    mat, partptr, perm = weighted_mat.partition(num_parts=2, recursive=True)
+    unweighted_mat = SparseTensor.from_dense(torch.ones((6, 6), device=device))
+    mat, partptr, perm = unweighted_mat.partition(num_parts=2, recursive=True)
     assert partptr.numel() == 3
     assert perm.numel() == 6
 
-    unweighted_mat = SparseTensor.from_dense(torch.ones((6, 6), device=device))
+    unweighted_mat = unweighted_mat.set_value(None)
     mat, partptr, perm = unweighted_mat.partition(num_parts=2, recursive=True)
     assert partptr.numel() == 3
     assert perm.numel() == 6