Merge branch 'main' into wfh/test_pending_writes_null

This commit is contained in:
William Fu-Hinthorn
2024-07-23 21:28:51 -07:00
5 changed files with 107 additions and 33 deletions
+1 -1
View File
@@ -123,7 +123,7 @@ class MemorySaverAssertCheckpointMetadata(MemorySaver):
class MemorySaverNoPending(MemorySaver):
def get_tuple(self, config: RunnableConfig) -> CheckpointTuple | None:
def get_tuple(self, config: RunnableConfig) -> Optional[CheckpointTuple]:
result = super().get_tuple(config)
if result:
return CheckpointTuple(result.config, result.checkpoint, result.metadata)