ci: Enable mypy checks for checkpoint-sqlite lib

This commit is contained in:
Nuno Campos
2024-09-19 08:46:44 -07:00
parent b8a8651c23
commit 6c0339ca4b
7 changed files with 72 additions and 53 deletions
@@ -31,7 +31,7 @@ _AIO_ERROR_MSG = (
)
class SqliteSaver(BaseCheckpointSaver):
class SqliteSaver(BaseCheckpointSaver[str]):
"""A checkpoint saver that stores checkpoints in a SQLite database.
Note:
@@ -487,6 +487,7 @@ class SqliteSaver(BaseCheckpointSaver):
config: RunnableConfig,
checkpoint: Checkpoint,
metadata: CheckpointMetadata,
new_versions: ChannelVersions,
) -> RunnableConfig:
"""Save a checkpoint to the database asynchronously.