diff --git a/queue_job/controllers/main.py b/queue_job/controllers/main.py index adc450d52..51f91794b 100644 --- a/queue_job/controllers/main.py +++ b/queue_job/controllers/main.py @@ -7,6 +7,7 @@ import time import traceback from io import StringIO +from typing import Optional from psycopg2 import OperationalError, errorcodes from werkzeug.exceptions import BadRequest, Forbidden @@ -27,7 +28,7 @@ class RunJobController(http.Controller): @classmethod - def _acquire_job(cls, env: api.Environment, job_uuid: str) -> Job | None: + def _acquire_job(cls, env: api.Environment, job_uuid: str) -> Optional[Job]: """Acquire a job for execution. - make sure it is in ENQUEUED state