FIX: fix UserWarning in get_norm_adj_mat

RUCAIBox · Apr 4, 2022 · 390c305 · 390c305
1 parent dbd5427
commit 390c305
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/recbole/model/general_recommender/ncl.py b/recbole/model/general_recommender/ncl.py
@@ -121,7 +121,7 @@ def get_norm_adj_mat(self):
         L = sp.coo_matrix(L)
         row = L.row
         col = L.col
-        i = torch.LongTensor([row, col])
+        i = torch.LongTensor(np.array([row, col]))
         data = torch.FloatTensor(L.data)
         SparseL = torch.sparse.FloatTensor(i, data, torch.Size(L.shape))
         return SparseL