From 947a233fc5f2694780c2bd0855ff01be06d33121 Mon Sep 17 00:00:00 2001 From: Nuno Campos Date: Tue, 8 Apr 2025 14:20:29 -0700 Subject: [PATCH] Fix --- libs/langgraph/tests/test_pregel_async.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/libs/langgraph/tests/test_pregel_async.py b/libs/langgraph/tests/test_pregel_async.py index 81f3c2289..dd146dbb7 100644 --- a/libs/langgraph/tests/test_pregel_async.py +++ b/libs/langgraph/tests/test_pregel_async.py @@ -5502,6 +5502,7 @@ async def test_subgraph_checkpoint_true( ] +@NEEDS_CONTEXTVARS @pytest.mark.parametrize("checkpoint_during", [True, False]) @pytest.mark.parametrize("checkpointer_name", REGULAR_CHECKPOINTERS_ASYNC) async def test_subgraph_checkpoint_true_interrupt( @@ -5534,7 +5535,7 @@ async def test_subgraph_checkpoint_true_interrupt( return {"foo": "hi! " + state["foo"]} async def node_2(state: ParentState, config: RunnableConfig): - response = await subgraph.ainvoke({"bar": state["foo"]}, config) + response = await subgraph.ainvoke({"bar": state["foo"]}) return {"foo": response["bar"]} builder = StateGraph(ParentState)