From 62dfa48ce57c46665eaa0aee9047a6e3db827016 Mon Sep 17 00:00:00 2001 From: Nuno Campos Date: Thu, 21 Mar 2024 17:58:23 -0700 Subject: [PATCH] Lint --- langgraph/graph/graph.py | 12 +++++------- 1 file changed, 5 insertions(+), 7 deletions(-) diff --git a/langgraph/graph/graph.py b/langgraph/graph/graph.py index 097e75af0..e3f15798b 100644 --- a/langgraph/graph/graph.py +++ b/langgraph/graph/graph.py @@ -150,12 +150,10 @@ class Graph: def set_finish_point(self, key: str) -> None: return self.add_edge(key, END) - def validate( - self, - interrupt: Optional[Sequence[str]] = None, - ) -> None: - edges = self._all_edges - all_starts = {src for src, _ in edges} | {src for src in self.branches} + def validate(self, interrupt: Optional[Sequence[str]] = None) -> None: + all_starts = {src for src, _ in self._all_edges} | { + src for src in self.branches + } for node in self.nodes: if node not in all_starts: raise ValueError(f"Node `{node}` is a dead-end") @@ -166,7 +164,7 @@ class Graph: if self.entry_point_branch is not None: branches.append(self.entry_point_branch) - all_hard_ends = {end for _, end in edges} + all_hard_ends = {end for _, end in self._all_edges} if self.entry_point is not None: all_hard_ends.add(self.entry_point)