This commit is contained in:
Sydney Runkle
2026-04-22 17:47:53 -04:00
parent 2e7edb2b60
commit 51154be4ab
3 changed files with 5 additions and 5 deletions
+2 -2
View File
@@ -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
+2 -2
View File
@@ -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
@@ -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 []