From 63324a337c5ad54caae8b720c0e505d83825604d Mon Sep 17 00:00:00 2001 From: Philipp Schlegel Date: Thu, 9 Jun 2022 10:54:10 +0100 Subject: [PATCH] fix Skeleton.get_graph --- skeletor/skeletonize/base.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/skeletor/skeletonize/base.py b/skeletor/skeletonize/base.py index e63efc7..ed7eaa2 100644 --- 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