From b9361d69b4eb64694d9078ddc8ead26a08e4afc4 Mon Sep 17 00:00:00 2001
From: Nuno Campos <nuno@langchain.dev>
Date: Wed, 4 Sep 2024 12:30:27 -0700
Subject: [PATCH] Try to fix test

---
 libs/langgraph/tests/test_pregel_async.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/libs/langgraph/tests/test_pregel_async.py b/libs/langgraph/tests/test_pregel_async.py
index 73de0b953..7afd292ae 100644
--- a/libs/langgraph/tests/test_pregel_async.py
+++ b/libs/langgraph/tests/test_pregel_async.py
@@ -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