Skip to content

Commit d0f5005

Browse files
committed
fix link
1 parent bb87ec6 commit d0f5005

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

torch_scatter/__init__.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,8 @@
1+
import os.path as osp
2+
13
import torch
24

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

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

0 commit comments

Comments
 (0)