Merge pull request #90 from rllm-team/yanjiashuo-patch-1-10

Update neighbor_sampler.py

rllm/sampler/neighbor_sampler.py

@@ -3,7 +3,9 @@
 
 import torch
 from torch import Tensor
+from torch_sparse import SparseTensor
 
+from rllm.utils.sparse import is_torch_sparse_tensor
 
 class NeighborSampler(torch.utils.data.DataLoader):
     r"""The neighbor sampler from the `"Inductive Representation Learning on