refactor(batch): use standard collections for types (#6475)

* Using generics types

* Using generics types

* Using generics types

* Using generics types

Author: leandrodamascena

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,

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
 

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