From 3d643965b5a9c73f05177e7adfbb76e9ff009850 Mon Sep 17 00:00:00 2001 From: Khalim Conn-Kowlessar Date: Tue, 7 Jul 2026 11:05:34 +0000 Subject: [PATCH] =?UTF-8?q?A=20failed=20batch=20sub=5Ftask=20restarts=20fo?= =?UTF-8?q?r=20a=20re-run=20=F0=9F=9F=A5?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Co-Authored-By: Claude Fable 5 --- tests/domain/tasks/test_subtasks.py | 14 ++++++++++++++ 1 file changed, 14 insertions(+) diff --git a/tests/domain/tasks/test_subtasks.py b/tests/domain/tasks/test_subtasks.py index 8cee44960..d379439c6 100644 --- a/tests/domain/tasks/test_subtasks.py +++ b/tests/domain/tasks/test_subtasks.py @@ -58,6 +58,20 @@ def test_start_rejects_from_terminal_status() -> None: st.start() +def test_start_restarts_a_failed_subtask_for_a_rerun() -> None: + # arrange + st = SubTask.create(task_id=uuid4()) + st.start() + st.fail(RuntimeError("degenerate prediction")) + + # act + st.start() + + # assert + assert st.status is SubTaskStatus.IN_PROGRESS + assert st.job_completed is None + + def test_complete_marks_outputs_and_job_completed() -> None: # arrange st = SubTask.create(task_id=uuid4())