Fix get neighbors with weights

biolab · Mar 20, 2024 · b9cd8ae · b9cd8ae
1 parent 76bebcb
commit b9cd8ae
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/orangecontrib/network/network/base.py b/orangecontrib/network/network/base.py
@@ -51,8 +51,7 @@ def _compose_neighbours(node: int, matrix: sp.csr_matrix, weights: bool):
         if not weights:
             return matrix.indices[fr:to]
         else:
-            return np.vstack(np.atleast_2d(matrix.indices[fr:to]),
-                             np.atleast_2d(matrix.data[fr:to]))
+            return matrix.indices[fr:to], matrix.data[fr:to]
 
     @staticmethod
     def _compute_degrees(edges, weighted):