Merge pull request #115 from rusty1s/travis

Travis

torch_scatter/segment_csr.py

-import warnings
+import importlib
 import os.path as osp
 from typing import Optional, Tuple
 
 import torch
 
-try:
-    torch.ops.load_library(
-        osp.join(osp.dirname(osp.abspath(__file__)), '_segment_csr.so'))
-except OSError:
-    warnings.warn('Failed to load `segment_csr` binaries.')
-
-    def segment_csr_placeholder(src: torch.Tensor, indptr: torch.Tensor,
-                                out: Optional[torch.Tensor]) -> torch.Tensor:
-        raise ImportError
-        return src
-
-    def segment_csr_with_arg_placeholder(
-            src: torch.Tensor, indptr: torch.Tensor,
-            out: Optional[torch.Tensor]) -> Tuple[torch.Tensor, torch.Tensor]:
-        raise ImportError
-        return src, indptr
-
-    def gather_csr_placeholder(src: torch.Tensor, indptr: torch.Tensor,
-                               out: Optional[torch.Tensor]) -> torch.Tensor:
-        raise ImportError
-        return src
-
-    torch.ops.torch_scatter.segment_sum_csr = segment_csr_placeholder
-    torch.ops.torch_scatter.segment_mean_csr = segment_csr_placeholder
-    torch.ops.torch_scatter.segment_min_csr = segment_csr_with_arg_placeholder
-    torch.ops.torch_scatter.segment_max_csr = segment_csr_with_arg_placeholder
-    torch.ops.torch_scatter.gather_csr = gather_csr_placeholder
+torch.ops.load_library(importlib.machinery.PathFinder().find_spec(
+    '_segment_csr', [osp.dirname(__file__)]).origin)
 
 
 @torch.jit.script