use typing.Self

keakon · May 22, 2024 · 615e6dd · 615e6dd
1 parent 5bfc2b3
commit 615e6dd
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/app/models/__init__.py b/app/models/__init__.py
@@ -1,4 +1,4 @@
-from typing import Any, Sequence, Type, TypeGuard, TypeVar
+from typing import Any, Self, Sequence, Type, TypeGuard, TypeVar
 
 from sqlalchemy import Column, Row
 from sqlalchemy.ext.asyncio import AsyncSession
@@ -59,7 +59,7 @@ async def get_by_ids(
         columns: list | tuple | InstrumentedAttribute | TextClause | Column | Mapped | None = None,
         for_update: bool = False,
         for_read: bool = False,
-    ) -> 'Sequence[BaseModel | Row]':
+    ) -> Sequence[Self | Row]:
         if not ids:
             return []
         if columns is None:
@@ -98,7 +98,7 @@ async def get_all(
         columns: list | tuple | InstrumentedAttribute | Mapped | None = None,
         for_update: bool = False,
         for_read: bool = False,
-    ) -> 'Sequence[BaseModel | Row]':
+    ) -> Sequence[Self | Row]:
         if columns is None:
             query = select(cls)
             scalar = True