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
|
||||
|
||||
|
||||
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
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue