fix Skeleton.get_graph

navis-org · Jun 9, 2022 · 63324a3 · Skylion007 · Jun 10, 2022 · 63324a3
1 parent 17dfe95
commit 63324a3
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/skeletor/skeletonize/base.py b/skeletor/skeletonize/base.py
@@ -158,7 +158,8 @@ def get_graph(self):
         dists = np.sqrt((dists ** 2).sum(axis=1))
 
         G = nx.DiGraph()
-        G.add_weighted_edges_from([(s, t, w) for s, t, w in zip(nodes, parents, dists)])
+        G.add_weighted_edges_from([(s, t, w) for s, t, w in zip(nodes.node_id.values,
+                                                                nodes.parent_id.values, dists)])
 
         return G