removed load_library calls

Author: rusty1s

torch_scatter/scatter.py

@@ -1,19 +1,9 @@
-import os
-import importlib
-import os.path as osp
 from typing import Optional, Tuple
 
 import torch
 
 from .utils import broadcast
 
-try:
-    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':
-        raise AttributeError(e)
-
 
 @torch.jit.script
 def scatter_sum(src: torch.Tensor, index: torch.Tensor, dim: int = -1,

torch_scatter/segment_coo.py

@@ -1,17 +1,7 @@
-import os
-import importlib
-import os.path as osp
 from typing import Optional, Tuple
 
 import torch
 
-try:
-    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':
-        raise AttributeError(e)
-
 
 @torch.jit.script
 def segment_sum_coo(src: torch.Tensor, index: torch.Tensor,

torch_scatter/segment_csr.py

@@ -1,17 +1,7 @@
-import os
-import importlib
-import os.path as osp
 from typing import Optional, Tuple
 
 import torch
 
-try:
-    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':
-        raise AttributeError(e)
-
 
 @torch.jit.script
 def segment_sum_csr(src: torch.Tensor, indptr: torch.Tensor,