Merge pull request #273 from janezd/fix-compose-neighbours

Fix neighbors/outgoing/ingoing with weights=False
biolab · Mar 22, 2024 · 6f8308b · 6f8308b
2 parents 846cf5e + b9cd8ae
commit 6f8308b
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):