Fix flaky test

langchain-ai · Sep 17, 2024 · da901b0 · da901b0
1 parent f03dc2f
commit da901b0
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/libs/langgraph/tests/test_pregel_async.py b/libs/langgraph/tests/test_pregel_async.py
@@ -7011,6 +7011,9 @@ async def outer_2(state: State):
         config = {"configurable": {"thread_id": "2"}}
         async for c in app.astream({"my_key": ""}, config, subgraphs=True):
             chunks.append((round(perf_counter() - start, 1), c))
+        for idx in range(len(chunks)):
+            elapsed, c = chunks[idx]
+            chunks[idx] = (round(elapsed - chunks[0][0], 1), c)
 
         assert chunks == [
             # arrives before "inner" finishes