fix(typing): fix typing of id_attribute fields (#318)

litestar-org · Dec 19, 2024 · 488711e · 488711e
1 parent a3164de
commit 488711e
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/advanced_alchemy/repository/_async.py b/advanced_alchemy/repository/_async.py
@@ -65,7 +65,7 @@
 class SQLAlchemyAsyncRepositoryProtocol(FilterableRepositoryProtocol[ModelT], Protocol[ModelT]):
     """Base Protocol"""
 
-    id_attribute: Any
+    id_attribute: str
     match_fields: list[str] | str | None = None
     statement: Select[tuple[ModelT]]
     session: AsyncSession | async_scoped_session[AsyncSession]
@@ -402,7 +402,7 @@ class SQLAlchemyAsyncRepository(SQLAlchemyAsyncRepositoryProtocol[ModelT], Filte
         :class:`~advanced_alchemy.repository._util.FilterableRepository`
     """
 
-    id_attribute: Any = "id"
+    id_attribute: str = "id"
     """Name of the unique identifier for the model."""
     loader_options: LoadSpec | None = None
     """Default loader options for the repository."""

diff --git a/advanced_alchemy/repository/_sync.py b/advanced_alchemy/repository/_sync.py
@@ -66,7 +66,7 @@
 class SQLAlchemySyncRepositoryProtocol(FilterableRepositoryProtocol[ModelT], Protocol[ModelT]):
     """Base Protocol"""
 
-    id_attribute: Any
+    id_attribute: str
     match_fields: list[str] | str | None = None
     statement: Select[tuple[ModelT]]
     session: Session | scoped_session[Session]
@@ -403,7 +403,7 @@ class SQLAlchemySyncRepository(SQLAlchemySyncRepositoryProtocol[ModelT], Filtera
         :class:`~advanced_alchemy.repository._util.FilterableRepository`
     """
 
-    id_attribute: Any = "id"
+    id_attribute: str = "id"
     """Name of the unique identifier for the model."""
     loader_options: LoadSpec | None = None
     """Default loader options for the repository."""