From cd6b086374d7dbf0b6e25d008e6c6e93095e4091 Mon Sep 17 00:00:00 2001
From: Sydney Runkle <sydneymarierunkle@gmail.com>
Date: Wed, 23 Apr 2025 16:36:37 -0700
Subject: [PATCH] await

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

diff --git a/libs/langgraph/tests/test_pregel_async.py b/libs/langgraph/tests/test_pregel_async.py
index df470195c..01e354b6b 100644
--- a/libs/langgraph/tests/test_pregel_async.py
+++ b/libs/langgraph/tests/test_pregel_async.py
@@ -6873,7 +6873,7 @@ async def test_interrupt_task_functional(checkpointer_name: str) -> None:
 
         config = {"configurable": {"thread_id": "1"}}
         # First run, interrupted at bar
-        assert graph.ainvoke({"a": ""}, config) == {
+        assert await graph.ainvoke({"a": ""}, config) == {
             "__interrupt__": [
                 Interrupt(
                     value="Provide value for bar:",