fix(db): generator types
This commit is contained in:
parent
fab228bf8f
commit
ee47186c5a
1 changed files with 2 additions and 2 deletions
|
|
@ -41,7 +41,7 @@ def get_db_connection() -> Generator[Connection, None, None]:
|
||||||
raise
|
raise
|
||||||
|
|
||||||
|
|
||||||
def _get_db_connection() -> Generator[Connection, None]:
|
def _get_db_connection() -> Generator[Connection, None, None]:
|
||||||
with get_db_connection() as connection:
|
with get_db_connection() as connection:
|
||||||
yield connection
|
yield connection
|
||||||
|
|
||||||
|
|
@ -61,7 +61,7 @@ def get_db_session() -> Generator[Session, None, None]:
|
||||||
session.close()
|
session.close()
|
||||||
|
|
||||||
|
|
||||||
def _get_db_session() -> Generator[Session, None]:
|
def _get_db_session() -> Generator[Session, None, None]:
|
||||||
with get_db_session() as session:
|
with get_db_session() as session:
|
||||||
yield session
|
yield session
|
||||||
|
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue