diff --git a/task_sdk/tests/execution_time/test_supervisor.py b/task_sdk/tests/execution_time/test_supervisor.py index f741d7c21b520..5ed51fece51aa 100644 --- a/task_sdk/tests/execution_time/test_supervisor.py +++ b/task_sdk/tests/execution_time/test_supervisor.py @@ -190,4 +190,4 @@ def subprocess_main(): assert proc.wait() == 0 assert spy.called_with(id, pid=proc.pid) # noqa: PGH005 # The exact number we get will depend on timing behaviour, so be a little lenient - assert 2 <= len(spy.calls) <= 4 + assert 1 <= len(spy.calls) <= 4