From 51154be4ab04f587a4dd98e60fdf1bd9f692a896 Mon Sep 17 00:00:00 2001 From: Sydney Runkle Date: Wed, 22 Apr 2026 17:47:53 -0400 Subject: [PATCH] lint --- libs/checkpoint-postgres/tests/test_async.py | 4 ++-- libs/checkpoint-postgres/tests/test_sync.py | 4 ++-- libs/checkpoint/langgraph/checkpoint/base/__init__.py | 2 +- 3 files changed, 5 insertions(+), 5 deletions(-) diff --git a/libs/checkpoint-postgres/tests/test_async.py b/libs/checkpoint-postgres/tests/test_async.py index 4a9f08336..9dc5c78cd 100644 --- a/libs/checkpoint-postgres/tests/test_async.py +++ b/libs/checkpoint-postgres/tests/test_async.py @@ -361,9 +361,9 @@ async def test_get_checkpoint_no_channel_values( load_checkpoint_tuple = saver._load_checkpoint_tuple - def patched_load_checkpoint_tuple(value): + async def patched_load_checkpoint_tuple(value, cur): value["checkpoint"].pop("channel_values", None) - return load_checkpoint_tuple(value) + return await load_checkpoint_tuple(value, cur) monkeypatch.setattr( saver, "_load_checkpoint_tuple", patched_load_checkpoint_tuple diff --git a/libs/checkpoint-postgres/tests/test_sync.py b/libs/checkpoint-postgres/tests/test_sync.py index 63b93715f..cc78c615e 100644 --- a/libs/checkpoint-postgres/tests/test_sync.py +++ b/libs/checkpoint-postgres/tests/test_sync.py @@ -348,9 +348,9 @@ def test_get_checkpoint_no_channel_values( load_checkpoint_tuple = saver._load_checkpoint_tuple - def patched_load_checkpoint_tuple(value): + def patched_load_checkpoint_tuple(value, cur): value["checkpoint"].pop("channel_values", None) - return load_checkpoint_tuple(value) + return load_checkpoint_tuple(value, cur) monkeypatch.setattr( saver, "_load_checkpoint_tuple", patched_load_checkpoint_tuple diff --git a/libs/checkpoint/langgraph/checkpoint/base/__init__.py b/libs/checkpoint/langgraph/checkpoint/base/__init__.py index 10382519e..f8fb8bbb7 100644 --- a/libs/checkpoint/langgraph/checkpoint/base/__init__.py +++ b/libs/checkpoint/langgraph/checkpoint/base/__init__.py @@ -516,7 +516,7 @@ class BaseCheckpointSaver(Generic[V]): self, config: RunnableConfig, channel: str ) -> List[Any]: # noqa: UP006 """Async version of get_channel_writes.""" - from langgraph.types import Overwrite # type: ignore[import-untyped] + from langgraph.types import Overwrite if getattr(_DELTA_RECONSTRUCTION, "active", False): return []