From 09c70d9eda07e87b3c07188452df2279121b30c8 Mon Sep 17 00:00:00 2001
From: Nuno Campos <nuno@langchain.dev>
Date: Thu, 13 Jun 2024 16:13:56 -0700
Subject: [PATCH] py 3.9

---
 tests/test_pregel_async.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/tests/test_pregel_async.py b/tests/test_pregel_async.py
index 839773fae..3517c0c13 100644
--- a/tests/test_pregel_async.py
+++ b/tests/test_pregel_async.py
@@ -3,7 +3,7 @@ import json
 import operator
 import time
 from collections import Counter
-from contextlib import aclosing, asynccontextmanager, contextmanager
+from contextlib import asynccontextmanager, contextmanager
 from typing import (
     Annotated,
     Any,
@@ -25,6 +25,7 @@ from langchain_core.runnables import (
     RunnablePassthrough,
     RunnablePick,
 )
+from langchain_core.utils.aiter import aclosing
 from pytest_mock import MockerFixture
 from syrupy import SnapshotAssertion