diff --git a/apps/worker/worker_app/tasks/cleanup.py b/apps/worker/worker_app/tasks/cleanup.py index f3f6903d8..28d5e7e22 100644 --- a/apps/worker/worker_app/tasks/cleanup.py +++ b/apps/worker/worker_app/tasks/cleanup.py @@ -7,7 +7,6 @@ import logging from celery import shared_task - from worker_app.tasks._startup import ( PENDING_TASK_TIMEOUT_MINUTES, cleanup_stale_pending_tasks, diff --git a/tests/unit/test_generation_task_pending_cleanup.py b/tests/unit/test_generation_task_pending_cleanup.py index ce257b937..252f1be19 100644 --- a/tests/unit/test_generation_task_pending_cleanup.py +++ b/tests/unit/test_generation_task_pending_cleanup.py @@ -37,6 +37,7 @@ def _make_task(**kwargs) -> GenerationTask: # cleanup_stale_pending 基本测试 # --------------------------------------------------------------------------- + def test_cleanup_stale_pending_no_tasks_returns_zero(): """没有任务时返回 0。""" repo, _, _ = _repository()