Skip to content

Commit 59a2d5b

Browse files
fix: typecheck
1 parent 26442ba commit 59a2d5b

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

packages/celery-library/src/celery_library/task_manager.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,6 +12,7 @@
1212
TASK_DONE_STATES,
1313
Task,
1414
TaskEvent,
15+
TaskEventID,
1516
TaskFilter,
1617
TaskID,
1718
TaskInfoStore,
@@ -193,7 +194,7 @@ async def consume_task_events(
193194
task_filter: TaskFilter,
194195
task_uuid: TaskUUID,
195196
last_id: str | None = None,
196-
) -> AsyncIterator[TaskEvent]:
197+
) -> AsyncIterator[tuple[TaskEventID, TaskEvent]]:
197198
task_id = task_filter.create_task_id(task_uuid=task_uuid)
198199
async for event in self._task_info_store.consume_task_events(
199200
task_id=task_id, last_id=last_id

0 commit comments

Comments
 (0)