diff --git a/aws_lambda_powertools/utilities/batch/base.py b/aws_lambda_powertools/utilities/batch/base.py index 2cb74296ca0..c70419873a1 100644 --- a/aws_lambda_powertools/utilities/batch/base.py +++ b/aws_lambda_powertools/utilities/batch/base.py @@ -14,7 +14,7 @@ import sys from abc import ABC, abstractmethod from enum import Enum -from typing import TYPE_CHECKING, Any, Callable, Tuple, Union, overload +from typing import TYPE_CHECKING, Any, Tuple, Union, overload from aws_lambda_powertools.shared import constants from aws_lambda_powertools.utilities.batch.exceptions import ( @@ -31,6 +31,8 @@ from aws_lambda_powertools.utilities.data_classes.sqs_event import SQSRecord if TYPE_CHECKING: + from collections.abc import Callable + from aws_lambda_powertools.utilities.batch.types import ( PartialItemFailureResponse, PartialItemFailures, diff --git a/aws_lambda_powertools/utilities/batch/decorators.py b/aws_lambda_powertools/utilities/batch/decorators.py index 2b9f5433e70..320535141fc 100644 --- a/aws_lambda_powertools/utilities/batch/decorators.py +++ b/aws_lambda_powertools/utilities/batch/decorators.py @@ -1,7 +1,7 @@ from __future__ import annotations import warnings -from typing import TYPE_CHECKING, Any, Awaitable, Callable +from typing import TYPE_CHECKING, Any from typing_extensions import deprecated @@ -16,6 +16,8 @@ from aws_lambda_powertools.warnings import PowertoolsDeprecationWarning if TYPE_CHECKING: + from collections.abc import Awaitable, Callable + from aws_lambda_powertools.utilities.batch.types import PartialItemFailureResponse from aws_lambda_powertools.utilities.typing import LambdaContext diff --git a/tests/functional/batch/required_dependencies/test_utilities_batch.py b/tests/functional/batch/required_dependencies/test_utilities_batch.py index 4c91dd54a1e..2a31c13946d 100644 --- a/tests/functional/batch/required_dependencies/test_utilities_batch.py +++ b/tests/functional/batch/required_dependencies/test_utilities_batch.py @@ -1,7 +1,9 @@ +from __future__ import annotations + import json import uuid from random import randint -from typing import Any, Awaitable, Callable, Dict +from typing import TYPE_CHECKING, Any import pytest @@ -26,6 +28,9 @@ from aws_lambda_powertools.warnings import PowertoolsDeprecationWarning from tests.functional.utils import b64_to_str, str_to_b64 +if TYPE_CHECKING: + from collections.abc import Awaitable, Callable + @pytest.fixture(scope="module") def sqs_event_fifo_factory() -> Callable: @@ -169,7 +174,7 @@ def handler(record: DynamoDBRecord): @pytest.fixture(scope="module") def order_event_factory() -> Callable: - def factory(item: Dict) -> str: + def factory(item: dict[str, Any]) -> str: return json.dumps({"item": item}) return factory