clean up

Author: rusty1s

.travis.yml

@@ -85,6 +85,6 @@ deploy:
   acl: public_read
   on:
     repo: rusty1s/pytorch_cluster
-    tags: true
+    condition: $TRAVIS_TAG != "" || $TRAVIS_COMMIT_MESSAGE =~ ci-deploy
 notifications:
   email: false

setup.py

@@ -63,7 +63,7 @@ def get_extensions():
 
 setup(
     name='torch_cluster',
-    version='1.5.3',
+    version='1.5.4',
     author='Matthias Fey',
     author_email='[email protected]',
     url='https://github.com/rusty1s/pytorch_cluster',

torch_cluster/__init__.py

@@ -3,24 +3,14 @@
 
 import torch
 
-__version__ = '1.5.3'
-expected_torch_version = (1, 4)
-
-try:
-    for library in [
-            '_version', '_grid', '_graclus', '_fps', '_rw', '_sampler',
-            '_nearest', '_knn', '_radius'
-    ]:
-        torch.ops.load_library(importlib.machinery.PathFinder().find_spec(
-            library, [osp.dirname(__file__)]).origin)
-except OSError as e:
-    major, minor = [int(x) for x in torch.__version__.split('.')[:2]]
-    t_major, t_minor = expected_torch_version
-    if major != t_major or (major == t_major and minor != t_minor):
-        raise RuntimeError(
-            f'Expected PyTorch version {t_major}.{t_minor} but found '
-            f'version {major}.{minor}.')
-    raise OSError(e)
+__version__ = '1.5.4'
+
+for library in [
+        '_version', '_grid', '_graclus', '_fps', '_rw', '_sampler', '_nearest',
+        '_knn', '_radius'
+]:
+    torch.ops.load_library(importlib.machinery.PathFinder().find_spec(
+        library, [osp.dirname(__file__)]).origin)
 
 if torch.version.cuda is not None:  # pragma: no cover
     cuda_version = torch.ops.torch_cluster.cuda_version()