reset

Author: rusty1s

.travis.yml

@@ -1,26 +1,26 @@
 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

@@ -1,7 +1,7 @@
 # flake8: noqa
 
 import os
-import glob
+import importlib
 import os.path as osp
 
 import torch
@@ -10,8 +10,8 @@
 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

@@ -1,5 +1,5 @@
 import os
-import glob
+import importlib
 import os.path as osp
 from typing import Optional, Tuple
 
@@ -8,8 +8,8 @@
 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

@@ -1,13 +1,13 @@
 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

@@ -1,13 +1,13 @@
 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