doc fix

Author: rusty1s

docs/requirements.txt

@@ -1,4 +1,3 @@
--f https://download.pytorch.org/whl/torch_stable.html
 numpy
 https://download.pytorch.org/whl/cpu/torch-1.4.0%2Bcpu-cp37-cp37m-linux_x86_64.whl
 sphinx

torch_scatter/__init__.py

@@ -22,19 +22,19 @@
                 f'version {major}.{minor}.')
     raise OSError(e)
 except AttributeError as e:
-    if os.getenv('BUILD_DOCS', '0') == '1':
-        pass
-    else:
+    if os.getenv('BUILD_DOCS', '0') != '1':
         raise AttributeError(e)
 
-cuda_version = torch.ops.torch_scatter.cuda_version()
-if cuda_version != -1 and torch.version.cuda is not None:  # pragma: no cover
-    if cuda_version < 10000:
+if torch.version.cuda is not None:  # pragma: no cover
+    cuda_version = torch.ops.torch_scatter.cuda_version()
+
+    if cuda_version == -1:
+        major = minor = 0
+    elif cuda_version < 10000:
         major, minor = int(str(cuda_version)[0]), int(str(cuda_version)[2])
     else:
         major, minor = int(str(cuda_version)[0:2]), int(str(cuda_version)[3])
     t_major, t_minor = [int(x) for x in torch.version.cuda.split('.')]
-    cuda_version = str(major) + '.' + str(minor)
 
     if t_major != major or t_minor != minor:
         raise RuntimeError(

torch_scatter/scatter.py

@@ -11,9 +11,7 @@
     torch.ops.load_library(importlib.machinery.PathFinder().find_spec(
         '_scatter', [osp.dirname(__file__)]).origin)
 except AttributeError as e:
-    if os.getenv('BUILD_DOCS', '0') == '1':
-        pass
-    else:
+    if os.getenv('BUILD_DOCS', '0') != '1':
         raise AttributeError(e)
 
 

torch_scatter/segment_coo.py

@@ -9,9 +9,7 @@
     torch.ops.load_library(importlib.machinery.PathFinder().find_spec(
         '_segment_coo', [osp.dirname(__file__)]).origin)
 except AttributeError as e:
-    if os.getenv('BUILD_DOCS', '0') == '1':
-        pass
-    else:
+    if os.getenv('BUILD_DOCS', '0') != '1':
         raise AttributeError(e)
 
 

torch_scatter/segment_csr.py

@@ -9,9 +9,7 @@
     torch.ops.load_library(importlib.machinery.PathFinder().find_spec(
         '_segment_csr', [osp.dirname(__file__)]).origin)
 except AttributeError as e:
-    if os.getenv('BUILD_DOCS', '0') == '1':
-        pass
-    else:
+    if os.getenv('BUILD_DOCS', '0') != '1':
         raise AttributeError(e)