diff --git a/sqs_workers/async_task.py b/sqs_workers/async_task.py index 43e6596..3a51b8f 100644 --- a/sqs_workers/async_task.py +++ b/sqs_workers/async_task.py @@ -8,6 +8,7 @@ Generator, Generic, NoReturn, + Optional, TypeVar, ) from typing_extensions import ParamSpec @@ -61,7 +62,7 @@ def batch(self) -> Generator[None, None, None]: with self.queue.add_batch(): yield - def delay(self, *args: P.args, **kwargs: P.kwargs) -> str | None: + def delay(self, *args: P.args, **kwargs: P.kwargs) -> Optional[str]: """ Run the task asynchronously. """ diff --git a/sqs_workers/queue.py b/sqs_workers/queue.py index ef96ddd..4ca571e 100644 --- a/sqs_workers/queue.py +++ b/sqs_workers/queue.py @@ -533,7 +533,7 @@ def add_raw_job( delay_seconds, deduplication_id, group_id: Optional[str], - ) -> str | None: + ) -> Optional[str]: """ Low-level function to put message to the queue """