This commit is contained in:
Nuno Campos
2024-03-13 17:42:14 -07:00
parent 53297db419
commit d169db951a
2 changed files with 8 additions and 8 deletions
+4 -4
View File
@@ -28,10 +28,10 @@ class AsyncSqliteSaver(BaseCheckpointSaver, AbstractAsyncContextManager):
async def __aexit__(
self,
__exc_type: type[BaseException] | None,
__exc_value: BaseException | None,
__traceback: TracebackType | None,
) -> bool | None:
__exc_type: Optional[type[BaseException]],
__exc_value: Optional[BaseException],
__traceback: Optional[TracebackType],
) -> Optional[bool]:
return await self.conn.close()
async def setup(self) -> None:
+4 -4
View File
@@ -28,10 +28,10 @@ class SqliteSaver(BaseCheckpointSaver, AbstractContextManager):
def __exit__(
self,
__exc_type: type[BaseException] | None,
__exc_value: BaseException | None,
__traceback: TracebackType | None,
) -> bool | None:
__exc_type: Optional[type[BaseException]],
__exc_value: Optional[BaseException],
__traceback: Optional[TracebackType],
) -> Optional[bool]:
return self.conn.close()
def setup(self) -> None: