remove sleep

This commit is contained in:
vbarda
2024-07-22 20:26:12 -04:00
parent aed313ebb5
commit cd4fae601c
2 changed files with 0 additions and 11 deletions
-1
View File
@@ -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):
-10
View File
@@ -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):