build: update pyright (#665)

Author: adhtruong

.pre-commit-config.yaml

@@ -52,7 +52,7 @@ repos:
             sqlalchemy>=2,
           ]
   - repo: https://github.com/RobertCraigie/pyright-python
-    rev: v1.1.394
+    rev: v1.1.397
     hooks:
       - id: pyright
         exclude: "tests"

polyfactory/factories/sqlalchemy_factory.py

@@ -1,7 +1,7 @@
 from __future__ import annotations
 
 from datetime import date, datetime
-from typing import TYPE_CHECKING, Any, Callable, ClassVar, Generic, List, TypeVar, Union
+from typing import TYPE_CHECKING, Any, Callable, ClassVar, Generic, List, Protocol, TypeVar, Union
 
 from typing_extensions import Annotated
 
@@ -68,6 +68,14 @@ async def save_many(self, data: list[T]) -> list[T]:
         return data
 
 
+_T_co = TypeVar("_T_co", covariant=True)
+
+
+class _SessionMaker(Protocol[_T_co]):
+    @staticmethod
+    def __call__() -> _T_co: ...
+
+
 class SQLAlchemyFactory(Generic[T], BaseFactory[T]):
     """Base factory for SQLAlchemy models."""
 
@@ -82,8 +90,8 @@ class SQLAlchemyFactory(Generic[T], BaseFactory[T]):
     __set_association_proxy__: ClassVar[bool] = False
     """Configuration to consider AssociationProxy property as a model field or not."""
 
-    __session__: ClassVar[Session | Callable[[], Session] | None] = None
-    __async_session__: ClassVar[AsyncSession | Callable[[], AsyncSession] | None] = None
+    __session__: ClassVar[Session | _SessionMaker[Session] | None] = None
+    __async_session__: ClassVar[AsyncSession | _SessionMaker[AsyncSession] | None] = None
 
     __config_keys__ = (
         *BaseFactory.__config_keys__,