fix geodesic_matrix when using networkx over igraph

navis-org · Jan 9, 2024 · a5d4b72 · a5d4b72
1 parent 9a1956d
commit a5d4b72
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/navis/graph/graph_utils.py b/navis/graph/graph_utils.py
@@ -682,6 +682,10 @@ def geodesic_matrix(x: 'core.NeuronObject',
         indices = None
         ix = nodeList
 
+    # For some reason csgrpah.dijkstra expects indices/indptr as int32
+    # igraph seems to do that by default but networkx uses int64 for indices
+    m.indptr = m.indptr.astype('int32', copy=False)
+    m.indices = m.indices.astype('int32', copy=False)
     dmat = csgraph.dijkstra(m,
                             directed=directed,
                             indices=indices,