1
0
Fork 0
forked from sr2/cloud-api

fix(db): generator types

This commit is contained in:
Iain Learmonth 2026-06-22 15:12:34 +01:00
parent fab228bf8f
commit ee47186c5a

View file

@ -41,7 +41,7 @@ def get_db_connection() -> Generator[Connection, None, None]:
raise
def _get_db_connection() -> Generator[Connection, None]:
def _get_db_connection() -> Generator[Connection, None, None]:
with get_db_connection() as connection:
yield connection
@ -61,7 +61,7 @@ def get_db_session() -> Generator[Session, None, None]:
session.close()
def _get_db_session() -> Generator[Session, None]:
def _get_db_session() -> Generator[Session, None, None]:
with get_db_session() as session:
yield session