This commit is contained in:
Nuno Campos
2024-06-14 13:35:28 -07:00
parent fe87c44da7
commit e8dc9f0678
2 changed files with 2 additions and 2 deletions
+1 -1
View File
@@ -180,7 +180,7 @@ def test_graph_validation() -> None:
def test_checkpoint_errors() -> None:
class FaultyGetCheckpointer(MemorySaver):
def get_tuple(self, config: RunnableConfig) -> CheckpointTuple | None:
def get_tuple(self, config: RunnableConfig) -> Optional[CheckpointTuple]:
raise ValueError("Faulty get_tuple")
class FaultyPutCheckpointer(MemorySaver):
+1 -1
View File
@@ -61,7 +61,7 @@ from tests.memory_assert import (
async def test_checkpoint_errors() -> None:
class FaultyGetCheckpointer(MemorySaver):
async def aget_tuple(self, config: RunnableConfig) -> CheckpointTuple | None:
async def aget_tuple(self, config: RunnableConfig) -> Optional[CheckpointTuple]:
raise ValueError("Faulty get_tuple")
class FaultyPutCheckpointer(MemorySaver):