Skip to content

Commit ac047fe

Browse files
chore(lint): fix lint in checkpointer tests (#299)
1 parent 4313ba2 commit ac047fe

File tree

2 files changed

+6
-6
lines changed

2 files changed

+6
-6
lines changed

tests/test_async_checkpoint.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -173,7 +173,7 @@ async def test_checkpoint_async(
173173

174174

175175
@pytest.fixture
176-
def test_data():
176+
def test_data() -> dict[str, Any]:
177177
"""Fixture providing test data for checkpoint tests."""
178178
config_0: RunnableConfig = {"configurable": {"thread_id": "1", "checkpoint_ns": ""}}
179179
config_1: RunnableConfig = {
@@ -224,13 +224,13 @@ def test_data():
224224
"source": "input",
225225
"step": 2,
226226
"writes": {},
227-
"parents": 1,
227+
"parents": 1, # type: ignore[typeddict-item]
228228
}
229229
metadata_2: CheckpointMetadata = {
230230
"source": "loop",
231231
"step": 1,
232232
"writes": {"foo": "bar"},
233-
"parents": None,
233+
"parents": None, # type: ignore[typeddict-item]
234234
}
235235
metadata_3: CheckpointMetadata = {}
236236

tests/test_checkpoint.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -197,7 +197,7 @@ def test_checkpoint_table(engine: Any) -> None:
197197

198198

199199
@pytest.fixture
200-
def test_data():
200+
def test_data() -> dict[str, Any]:
201201
"""Fixture providing test data for checkpoint tests."""
202202
config_0: RunnableConfig = {"configurable": {"thread_id": "1", "checkpoint_ns": ""}}
203203
config_1: RunnableConfig = {
@@ -248,13 +248,13 @@ def test_data():
248248
"source": "input",
249249
"step": 2,
250250
"writes": {},
251-
"parents": 1,
251+
"parents": 1, # type: ignore[typeddict-item]
252252
}
253253
metadata_2: CheckpointMetadata = {
254254
"source": "loop",
255255
"step": 1,
256256
"writes": {"foo": "bar"},
257-
"parents": None,
257+
"parents": None, # type: ignore[typeddict-item]
258258
}
259259
metadata_3: CheckpointMetadata = {}
260260

0 commit comments

Comments
 (0)