From 0ac29434a7e28351e7164a22988dd0dfa2f273ec Mon Sep 17 00:00:00 2001 From: Nuno Campos Date: Thu, 20 Mar 2025 08:40:05 -0700 Subject: [PATCH] Lint --- libs/langgraph/langgraph/pregel/loop.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/libs/langgraph/langgraph/pregel/loop.py b/libs/langgraph/langgraph/pregel/loop.py index 8f3331349..94faff87f 100644 --- a/libs/langgraph/langgraph/pregel/loop.py +++ b/libs/langgraph/langgraph/pregel/loop.py @@ -582,7 +582,7 @@ class PregelLoop(LoopProtocol): else: task.writes.append((k, v)) - def _first(self, *, input_keys: Union[str, Sequence[str]]) -> Union[set[str]]: + def _first(self, *, input_keys: Union[str, Sequence[str]]) -> Optional[set[str]]: # resuming from previous checkpoint requires # - finding a previous checkpoint # - receiving None input (outer graph) or RESUMING flag (subgraph) @@ -600,7 +600,7 @@ class PregelLoop(LoopProtocol): ) ) # this can be set only when there are input_writes - updated_channels: Union[set[str]] = None + updated_channels: Optional[set[str]] = None # map command to writes if isinstance(self.input, Command):