source and target should be tuples in edge dict

talmolab · Sep 13, 2024 · 69db671 · 69db671
1 parent 71b0307
commit 69db671
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/sleap/skeleton.py b/sleap/skeleton.py
@@ -1052,13 +1052,12 @@ def to_json(self, node_to_idx: Optional[Dict[Node, int]] = None) -> str:
                     "key": 0,  # Always 0.
                     "source": {
                         "py/object": "sleap.skeleton.Node",
-                        "py/state": {"name": source.name, "weight": 1.0},
+                        "py/state": {"py/tuple": [source.name, source.weight]},
                     },
                     "target": {
                         "py/object": "sleap.skeleton.Node",
-                        "py/state": {"name": target.name, "weight": 1.0},
+                        "py/state": {"py/tuple": [target.name, target.weight]},
                     },
-                    # "target": {"py/id": node_to_id[target]},
                     "type": edge_type,
                 }
             )