Fix AsyncSession.exec return type

fastapi · Oct 12, 2023 · f83088c · f83088c
1 parent 30c5a4c
commit f83088c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sqlmodel/ext/asyncio/session.py b/sqlmodel/ext/asyncio/session.py
@@ -79,7 +79,7 @@ async def exec(
         execution_options: Mapping[Any, Any] = util.EMPTY_DICT,
         bind_arguments: Optional[Mapping[str, Any]] = None,
         **kw: Any,
-    ) -> ScalarResult[_TSelectParam]:
+    ) -> Union[Result[_TSelectParam], ScalarResult[_TSelectParam]]:
         # TODO: the documentation says execution_options accepts a dict, but only
         # util.immutabledict has the union() method. Is this a bug in SQLAlchemy?
         execution_options = execution_options.union({"prebuffer_rows": True})  # type: ignore