File tree Expand file tree Collapse file tree 1 file changed +3
-1
lines changed Expand file tree Collapse file tree 1 file changed +3
-1
lines changed Original file line number Diff line number Diff line change 77
88import firebolt .async_db as async_dbapi
99from firebolt .async_db import Connection
10+ from sqlalchemy .engine import AdaptedConnection # type: ignore[attr-defined]
1011
1112# Ignoring type since sqlalchemy-stubs doesn't cover AdaptedConnection
1213# and util.concurrency
13- from sqlalchemy .engine import AdaptedConnection # type: ignore[attr-defined]
14+ from sqlalchemy .pool import AsyncAdaptedQueuePool # type: ignore[attr-defined]
1415from sqlalchemy .util .concurrency import await_only # type: ignore[import]
1516from trio import run
1617
@@ -167,6 +168,7 @@ class AsyncFireboltDialect(FireboltDialect):
167168 supports_statement_cache : bool = False
168169 supports_server_side_cursors : bool = False
169170 is_async : bool = True
171+ poolclass = AsyncAdaptedQueuePool
170172
171173 @classmethod
172174 def dbapi (cls ) -> AsyncAPIWrapper :
You can’t perform that action at this time.
0 commit comments