From 2b4d7d8897cc580fcef74f59d4e517d163607783 Mon Sep 17 00:00:00 2001
From: yanjiashuo <59167722+yanjiashuo@users.noreply.github.com>
Date: Sun, 1 Sep 2024 22:01:34 +0800
Subject: [PATCH] Update neighbor_sampler.py

---
 rllm/sampler/neighbor_sampler.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/rllm/sampler/neighbor_sampler.py b/rllm/sampler/neighbor_sampler.py
index dee3c5a4..d6a30123 100644
--- 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):