From ed7f038a190f67b7304b3d1a609b249caae56f91 Mon Sep 17 00:00:00 2001 From: Nuno Campos Date: Fri, 9 May 2025 10:54:12 -0700 Subject: [PATCH] Fix --- libs/langgraph/tests/test_checkpoint_migration.py | 13 ++++--------- 1 file changed, 4 insertions(+), 9 deletions(-) diff --git a/libs/langgraph/tests/test_checkpoint_migration.py b/libs/langgraph/tests/test_checkpoint_migration.py index caba88060..f6583cada 100644 --- a/libs/langgraph/tests/test_checkpoint_migration.py +++ b/libs/langgraph/tests/test_checkpoint_migration.py @@ -1672,16 +1672,11 @@ async def test_latest_checkpoint_state_graph_async( @NEEDS_CONTEXTVARS @pytest.mark.parametrize("checkpoint_version", ["3", "2-start:*", "2-quadratic"]) def test_saved_checkpoint_state_graph( - request: pytest.FixtureRequest, - checkpointer_name: str, + sync_checkpointer: BaseCheckpointSaver, checkpoint_version: str, ) -> None: - checkpointer: BaseCheckpointSaver = request.getfixturevalue( - f"checkpointer_{checkpointer_name}" - ) - builder = make_state_graph() - app = builder.compile(checkpointer=checkpointer) + app = builder.compile(checkpointer=sync_checkpointer) thread1 = "1" config = {"configurable": {"thread_id": thread1, "checkpoint_ns": ""}} @@ -1693,8 +1688,8 @@ def test_saved_checkpoint_state_graph( for write in checkpoint.pending_writes: grouped_writes[write[0]].append(write[1:]) for tid, group in grouped_writes.items(): - checkpointer.put_writes(checkpoint.config, group, tid) - checkpointer.put( + sync_checkpointer.put_writes(checkpoint.config, group, tid) + sync_checkpointer.put( patch_configurable(config, {"checkpoint_id": parent_id}), checkpoint.checkpoint, checkpoint.metadata,