fix link

d0f5005f · rusty1s · bb87ec65 · d0f5005f
Commit d0f5005f authored Jan 29, 2020 by rusty1s
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 1 deletion

torch_scatter/__init__.py torch_scatter/__init__.py +3 -1

No files found.
--- a/torch_scatter/__init__.py
+++ b/torch_scatter/__init__.py
+import os.path as osp
+
 import torch

-torch.ops.load_library('torch_scatter/_C.so')
+torch.ops.load_library(osp.join(osp.dirname(osp.abspath(__file__)), '_C.so'))

 from .segment_csr import (segment_sum_csr, segment_add_csr, segment_mean_csr,
                          segment_min_csr, segment_max_csr, segment_csr,