remember edge type when unfusing

zxcalc · Nov 18, 2023 · 989d32c · 989d32c
1 parent 98f9643
commit 989d32c
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/zxlive/custom_rule.py b/zxlive/custom_rule.py
@@ -112,8 +112,9 @@ def unfuse_zx(self, graph, subgraph_nx, v, vtype):
         graph.add_edge(graph.edge(new_v, v))
         for b in neighbors:
             if b not in subgraph_nx.nodes:
+                graph.add_edge((new_v, b), graph.edge_type((v, b)))
                 graph.remove_edge(graph.edge(v, b))
-                graph.add_edge(graph.edge(new_v, b))
+
 
     def matcher(self, graph: GraphT, in_selection: Callable[[VT], bool]) -> list[VT]:
         vertices = [v for v in graph.vertices() if in_selection(v)]