Try to fix test

This commit is contained in:
Nuno Campos
2024-09-04 12:30:27 -07:00
parent 26b4d9139e
commit b9361d69b4
@@ -514,6 +514,7 @@ async def test_cancel_graph_astream_events_v2(checkpointer_name: Optional[str])
if chunk["event"] == "on_chain_stream" and not chunk["parent_ids"]:
got_event = True
assert chunk["data"]["chunk"] == {"alittlewhile": {"value": 2}}
await asyncio.sleep(0)
break
# did break