diff --git a/src/murfey/server/ispyb.py b/src/murfey/server/ispyb.py index be0536a9f..154067eaa 100644 --- a/src/murfey/server/ispyb.py +++ b/src/murfey/server/ispyb.py @@ -2,7 +2,7 @@ import datetime import logging -from typing import Callable, List, Literal, Optional +from typing import Callable, Generator, List, Literal, Optional import ispyb @@ -535,7 +535,7 @@ def do_buffer_lookup(self, app_id: int, uuid: int) -> Optional[int]: return reference -def _get_session() -> sqlalchemy.orm.Session: +def _get_session() -> Generator[Optional[sqlalchemy.orm.Session], None, None]: db = Session() if db is None: yield None