Merge pull request #84 from rllm-team/yanjiashuo-patch-2-3

Update neighbor_sampler.py
rllm-team · Sep 1, 2024 · 1eff54f · 1eff54f
2 parents 9a5fd0e + 2b4d7d8
commit 1eff54f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/rllm/sampler/neighbor_sampler.py b/rllm/sampler/neighbor_sampler.py
@@ -3,9 +3,9 @@
 
 import torch
 from torch import Tensor
-from torch_sparse import SparseTensor
+#from torch_sparse import SparseTensor
 
-from rllm.utils.sparse import is_torch_sparse_tensor
+#from rllm.utils.sparse import is_torch_sparse_tensor
 
 
 class NeighborSampler(torch.utils.data.DataLoader):