fix broken interrupt tests

This commit is contained in:
vbarda
2024-07-22 12:18:12 -04:00
parent d635a22302
commit f2a95ad67f
2 changed files with 2 additions and 0 deletions
+1
View File
@@ -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):
@@ -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):