diff --git a/libs/langgraph/tests/test_pregel.py b/libs/langgraph/tests/test_pregel.py index ee6545120..b0a822209 100644 --- a/libs/langgraph/tests/test_pregel.py +++ b/libs/langgraph/tests/test_pregel.py @@ -8457,7 +8457,6 @@ 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 d520c4092..d5cdfc472 100644 --- a/libs/langgraph/tests/test_pregel_async.py +++ b/libs/langgraph/tests/test_pregel_async.py @@ -6145,7 +6145,6 @@ async def test_nested_graph_interrupts( assert await app.ainvoke({"my_key": "my value"}, config, debug=True) == { "my_key": "hi my value", } - await asyncio.sleep(0.05) assert [s async for s in app.aget_state_history(config)] == [ StateSnapshot( values={"my_key": "hi my value"}, @@ -6187,7 +6186,6 @@ async def test_nested_graph_interrupts( assert await app.ainvoke(None, config, debug=True) == { "my_key": "hi my value here and there and back again", } - await asyncio.sleep(0.05) assert [s async for s in app.aget_state_history(config)] == [ StateSnapshot( values={"my_key": "hi my value here and there and back again"}, @@ -6324,7 +6322,6 @@ async def test_nested_graph_interrupts( "my_key": "hi my value", }, ] - await asyncio.sleep(0.05) assert [s async for s in app.aget_state_history(config)] == [ StateSnapshot( values={"my_key": "hi my value"}, @@ -6365,7 +6362,6 @@ async def test_nested_graph_interrupts( ] # while we're waiting for the node w/ interrupt inside to finish assert [c async for c in app.astream(None, config, stream_mode="values")] == [] - await asyncio.sleep(0.05) assert [s async for s in app.aget_state_history(config)] == [ StateSnapshot( values={"my_key": "hi my value"}, @@ -6412,7 +6408,6 @@ async def test_nested_graph_interrupts( "my_key": "hi my value here and there and back again", }, ] - await asyncio.sleep(0.05) assert [s async for s in app.aget_state_history(config)] == [ StateSnapshot( values={"my_key": "hi my value here and there and back again"}, @@ -6516,7 +6511,6 @@ async def test_nested_graph_interrupts( "my_key": "hi my value", }, ] - await asyncio.sleep(0.05) assert [s async for s in app.aget_state_history(config)] == [ StateSnapshot( values={"my_key": "hi my value"}, @@ -6560,7 +6554,6 @@ async def test_nested_graph_interrupts( "my_key": "hi my value here and there", }, ] - await asyncio.sleep(0.05) assert [s async for s in app.aget_state_history(config)] == [ StateSnapshot( values={"my_key": "hi my value here and there"}, @@ -6626,7 +6619,6 @@ async def test_nested_graph_interrupts( "my_key": "hi my value here and there and back again", }, ] - await asyncio.sleep(0.05) assert [s async for s in app.aget_state_history(config)] == [ StateSnapshot( values={"my_key": "hi my value here and there and back again"}, @@ -6759,7 +6751,6 @@ async def test_nested_graph_interrupts( ), ] - await asyncio.sleep(0.05) child_state_history = [ c async for c in app.aget_state_history( @@ -6993,7 +6984,6 @@ 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):