Skip to content

Commit ba8974a

Browse files
committed
fix tests
1 parent b802de0 commit ba8974a

File tree

2 files changed

+6
-2
lines changed

2 files changed

+6
-2
lines changed

tests/test_async_checkpoint.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -78,6 +78,7 @@
7878
"__start__": {"__start__": 1},
7979
"node": {"start:node": 2},
8080
},
81+
"updated_channels": [],
8182
}
8283

8384

@@ -212,6 +213,7 @@ def test_data() -> dict[str, Any]:
212213
"__start__": {"__start__": 1},
213214
"node": {"start:node": 2},
214215
},
216+
"updated_channels": [],
215217
}
216218
chkpnt_1: Checkpoint = empty_checkpoint()
217219
chkpnt_2: Checkpoint = create_checkpoint(chkpnt_1, {}, 1)
@@ -423,4 +425,4 @@ async def test_metadata(
423425
0
424426
].metadata[
425427
"my_key" # type: ignore
426-
] == "abc" # type: ignore
428+
] == "abc" # type: ignore

tests/test_checkpoint.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -59,6 +59,7 @@
5959
"__start__": {"__start__": 1},
6060
"node": {"start:node": 2},
6161
},
62+
"updated_channels": [],
6263
}
6364

6465

@@ -235,6 +236,7 @@ def test_data() -> dict[str, Any]:
235236
"__start__": {"__start__": 1},
236237
"node": {"start:node": 2},
237238
},
239+
"updated_channels": [],
238240
}
239241
chkpnt_1: Checkpoint = empty_checkpoint()
240242
chkpnt_2: Checkpoint = create_checkpoint(chkpnt_1, {}, 1)
@@ -348,4 +350,4 @@ def test_checkpoint_get_tuple(
348350
assert search_results_1.metadata == metadata[0] # type: ignore
349351

350352
# No matching checkpoint
351-
assert checkpointer.get_tuple(configs[0]) is None
353+
assert checkpointer.get_tuple(configs[0]) is None

0 commit comments

Comments
 (0)