Draw end edge for cond edge without mapping

2026-09-07 18:27:52 +02:00 · 2024-02-29 18:26:28 -08:00
parent 13f08e9bff
commit 2a624274fd
1 changed files with 4 additions and 1 deletions
@@ -257,7 +257,10 @@ class CompiledGraph(Pregel):
                    name,
                )
                graph.add_edge(graph.nodes[start], graph.nodes[name])
-                ends = branch.ends or {k: k for k in self.graph.nodes}
+                ends = branch.ends or {
+                    **{k: k for k in self.graph.nodes},
+                    END: END,
+                }
                for label, end in ends.items():
                    graph.add_edge(graph.nodes[name], graph.nodes[end], label)
        if self.graph.entry_point_branch: