merge

libs/checkpoint-postgres/tests/test_async_store.py

@@ -1,7 +1,7 @@
 # type: ignore
 import sys
 import uuid
-from typing import AsyncIterator
+from collections.abc import AsyncIterator
 
 import pytest
 from conftest import DEFAULT_URI  # type: ignore
@@ -43,9 +43,8 @@ async def store(request) -> AsyncIterator[AsyncPostgresStore]:
                 yield store
         elif request.param == "pool":
             async with AsyncPostgresStore.from_conn_string(
-                conn_string, use_pool=True, max_size=10
+                conn_string, pool_config={"min_size": 1, "max_size": 10}
             ) as store:
-                await store.setup()
                 yield store
         else:  # default
             async with AsyncPostgresStore.from_conn_string(conn_string) as store: