error handling

torch_scatter/__init__.py

@@ -12,13 +12,12 @@ try:
         torch.ops.load_library(importlib.machinery.PathFinder().find_spec(
             library, [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]]
-        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}.')
+    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)
 except AttributeError as e:
     if os.getenv('BUILD_DOCS', '0') != '1':