Fix tests

Author: frostedoyster

tests/test_init.py

@@ -1,5 +1,4 @@
-from pet_neighbors_convert import load_neighbors_convert
-
-
 def test_init():
-    load_neighbors_convert()
+    import torch
+    import pet_neighbors_convert
+    torch.ops.neighbors_convert.process