Fix plotting for using sleap-io API (#47)

talmolab · Jul 26, 2023 · c565235 · c565235
1 parent 02ee4e3
commit c565235
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/sleap_roots/series.py b/sleap_roots/series.py
@@ -277,9 +277,9 @@ def plot_instance(
             h_lines.append(h_lines_k)
 
     else:
-        for k, (src_node, dst_node) in enumerate(skeleton.edges):
-            src_pt = instance.points_array[instance.skeleton.node_to_index(src_node)]
-            dst_pt = instance.points_array[instance.skeleton.node_to_index(dst_node)]
+        for k, (src_ind, dst_ind) in enumerate(skeleton.edge_inds):
+            src_pt = inst_pts[src_ind]
+            dst_pt = inst_pts[dst_ind]
 
             x = np.array([src_pt[0], dst_pt[0]])
             y = np.array([src_pt[1], dst_pt[1]])