reset

.travis.yml

 language: shell
 
 os:
-  # - linux
-  # - osx
+  - linux
+  - osx
   - windows
 
 env:
   global:
     - CUDA_HOME=/usr/local/cuda
   jobs:
-    # - TORCH_VERSION=1.4.0 PYTHON_VERSION=3.8 IDX=cpu
-    # - TORCH_VERSION=1.4.0 PYTHON_VERSION=3.8 IDX=cu92
-    # - TORCH_VERSION=1.4.0 PYTHON_VERSION=3.8 IDX=cu100
+    - TORCH_VERSION=1.4.0 PYTHON_VERSION=3.8 IDX=cpu
+    - TORCH_VERSION=1.4.0 PYTHON_VERSION=3.8 IDX=cu92
+    - TORCH_VERSION=1.4.0 PYTHON_VERSION=3.8 IDX=cu100
     - TORCH_VERSION=1.4.0 PYTHON_VERSION=3.8 IDX=cu101
-    # - TORCH_VERSION=1.4.0 PYTHON_VERSION=3.7 IDX=cpu
-    # - TORCH_VERSION=1.4.0 PYTHON_VERSION=3.7 IDX=cu92
-    # - TORCH_VERSION=1.4.0 PYTHON_VERSION=3.7 IDX=cu100
-    # - TORCH_VERSION=1.4.0 PYTHON_VERSION=3.7 IDX=cu101
-    # - TORCH_VERSION=1.4.0 PYTHON_VERSION=3.6 IDX=cpu
-    # - TORCH_VERSION=1.4.0 PYTHON_VERSION=3.6 IDX=cu92
-    # - TORCH_VERSION=1.4.0 PYTHON_VERSION=3.6 IDX=cu100
-    # - TORCH_VERSION=1.4.0 PYTHON_VERSION=3.6 IDX=cu101
+    - TORCH_VERSION=1.4.0 PYTHON_VERSION=3.7 IDX=cpu
+    - TORCH_VERSION=1.4.0 PYTHON_VERSION=3.7 IDX=cu92
+    - TORCH_VERSION=1.4.0 PYTHON_VERSION=3.7 IDX=cu100
+    - TORCH_VERSION=1.4.0 PYTHON_VERSION=3.7 IDX=cu101
+    - TORCH_VERSION=1.4.0 PYTHON_VERSION=3.6 IDX=cpu
+    - TORCH_VERSION=1.4.0 PYTHON_VERSION=3.6 IDX=cu92
+    - TORCH_VERSION=1.4.0 PYTHON_VERSION=3.6 IDX=cu100
+    - TORCH_VERSION=1.4.0 PYTHON_VERSION=3.6 IDX=cu101
 
 jobs:
   exclude:  # Exclude *all* macOS CUDA jobs and Windows CUDA 9.2/10.0 jobs.

torch_scatter/__init__.py

 # flake8: noqa
 
 import os
-import glob
+import importlib
 import os.path as osp
 
 import torch
@@ -10,8 +10,8 @@ __version__ = '2.0.3'
 expected_torch_version = (1, 4)
 
 try:
-    torch.ops.load_library(
-        glob.glob(osp.join(osp.dirname(__file__), '_version.*'))[0])
+    torch.ops.load_library(importlib.machinery.PathFinder().find_spec(
+        '_version', [osp.dirname(__file__)]).origin)
 except OSError as e:
     if 'undefined symbol' in str(e):
         major, minor = [int(x) for x in torch.__version__.split('.')[:2]]

torch_scatter/scatter.py

 import os
-import glob
+import importlib
 import os.path as osp
 from typing import Optional, Tuple
 
@@ -8,8 +8,8 @@ import torch
 from .utils import broadcast
 
 try:
-    torch.ops.load_library(
-        glob.glob(osp.join(osp.dirname(__file__), '_scatter.*'))[0])
+    torch.ops.load_library(importlib.machinery.PathFinder().find_spec(
+        '_scatter', [osp.dirname(__file__)]).origin)
 except OSError as e:
     if os.getenv('BUILD_DOCS', '0') == '1':
         pass

torch_scatter/segment_coo.py

 import os
-import glob
+import importlib
 import os.path as osp
 from typing import Optional, Tuple
 
 import torch
 
 try:
-    torch.ops.load_library(
-        glob.glob(osp.join(osp.dirname(__file__), '_segment_coo.*'))[0])
+    torch.ops.load_library(importlib.machinery.PathFinder().find_spec(
+        '_segment_coo', [osp.dirname(__file__)]).origin)
 except OSError as e:
     if os.getenv('BUILD_DOCS', '0') == '1':
         pass

torch_scatter/segment_csr.py

 import os
-import glob
+import importlib
 import os.path as osp
 from typing import Optional, Tuple
 
 import torch
 
 try:
-    torch.ops.load_library(
-        glob.glob(osp.join(osp.dirname(__file__), '_segment_csr.*'))[0])
+    torch.ops.load_library(importlib.machinery.PathFinder().find_spec(
+        '_segment_csr', [osp.dirname(__file__)]).origin)
 except OSError as e:
     if os.getenv('BUILD_DOCS', '0') == '1':
         pass