Update async_sqlalchemy.py

uriyyo · Jul 29, 2021 · 95fbd72 · 95fbd72
1 parent 9577558
commit 95fbd72
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/fastapi_pagination/ext/async_sqlalchemy.py b/fastapi_pagination/ext/async_sqlalchemy.py
@@ -10,17 +10,17 @@
 
 if TYPE_CHECKING:
     from sqlalchemy.ext.asyncio import AsyncSession
-    from sqlalchemy.orm import Query
+    from sqlalchemy.sql import Select
 
 
 async def paginate(
     session: AsyncSession,
-    query: Query,
+    query: Select,
     params: Optional[AbstractParams] = None,
 ) -> AbstractPage:  # pragma: no cover # FIXME: fix coverage report generation
     params = resolve_params(params)
 
-    total = await session.scalar(select(func.count()).select_from(query.subquery()))
+    total = await session.scalar(select(func.count()).select_from(query.subquery()))  # type: ignore
     items = await session.execute(paginate_query(query, params))
 
     return create_page([*items.scalars()], total, params)