From 1c48ec0abaaff525a8ade92a9f06204091d54f2f Mon Sep 17 00:00:00 2001 From: Nuno Campos Date: Wed, 2 Apr 2025 12:48:10 -0700 Subject: [PATCH] Lint --- libs/langgraph/langgraph/pregel/__init__.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/libs/langgraph/langgraph/pregel/__init__.py b/libs/langgraph/langgraph/pregel/__init__.py index 8e2aa812f..545c8faf1 100644 --- a/libs/langgraph/langgraph/pregel/__init__.py +++ b/libs/langgraph/langgraph/pregel/__init__.py @@ -1233,7 +1233,8 @@ class Pregel(PregelProtocol): # get last checkpoint config = ensure_config(self.config, input_config) saved = checkpointer.get_tuple(config) - self._migrate_checkpoint(saved.checkpoint) + if saved is not None: + self._migrate_checkpoint(saved.checkpoint) checkpoint = ( copy_checkpoint(saved.checkpoint) if saved else empty_checkpoint() ) @@ -1644,7 +1645,8 @@ class Pregel(PregelProtocol): # get last checkpoint config = ensure_config(self.config, input_config) saved = await checkpointer.aget_tuple(config) - self._migrate_checkpoint(saved.checkpoint) + if saved is not None: + self._migrate_checkpoint(saved.checkpoint) checkpoint = ( copy_checkpoint(saved.checkpoint) if saved else empty_checkpoint() )