From f2a95ad67f73099f9eeb26989af501c1d0765dd3 Mon Sep 17 00:00:00 2001 From: vbarda Date: Mon, 22 Jul 2024 12:18:12 -0400 Subject: [PATCH] fix broken interrupt tests --- libs/langgraph/tests/test_pregel.py | 1 + libs/langgraph/tests/test_pregel_async.py | 1 + 2 files changed, 2 insertions(+) diff --git a/libs/langgraph/tests/test_pregel.py b/libs/langgraph/tests/test_pregel.py index f6f1257aa..1a2314a48 100644 --- a/libs/langgraph/tests/test_pregel.py +++ b/libs/langgraph/tests/test_pregel.py @@ -7687,6 +7687,7 @@ def test_nested_graph_interrupts_parallel(checkpointer: BaseCheckpointSaver) -> my_other_key: str def inner_1(state: InnerState): + time.sleep(0.1) return {"my_key": "got here", "my_other_key": state["my_key"]} def inner_2(state: InnerState): diff --git a/libs/langgraph/tests/test_pregel_async.py b/libs/langgraph/tests/test_pregel_async.py index 7e60c77fb..b168a7f08 100644 --- a/libs/langgraph/tests/test_pregel_async.py +++ b/libs/langgraph/tests/test_pregel_async.py @@ -6196,6 +6196,7 @@ async def test_nested_graph_interrupts_parallel( my_other_key: str async def inner_1(state: InnerState): + await asyncio.sleep(0.1) return {"my_key": "got here", "my_other_key": state["my_key"]} async def inner_2(state: InnerState):