chore: fmt and linting

Author: heitorlessa

aws_lambda_powertools/utilities/batch/__init__.py

@@ -5,15 +5,15 @@
 """
 
 from aws_lambda_powertools.utilities.batch.base import (
-    BasePartialProcessor,
+    AsyncBatchProcessor,
     BasePartialBatchProcessor,
+    BasePartialProcessor,
     BatchProcessor,
-    AsyncBatchProcessor,
     EventType,
     FailureResponse,
     SuccessResponse,
-    batch_processor,
     async_batch_processor,
+    batch_processor,
 )
 from aws_lambda_powertools.utilities.batch.exceptions import ExceptionInfo
 

aws_lambda_powertools/utilities/batch/base.py

@@ -11,7 +11,18 @@
 import sys
 from abc import ABC, abstractmethod
 from enum import Enum
-from typing import Any, Callable, Dict, List, Optional, Tuple, Type, Union, overload, Awaitable
+from typing import (
+    Any,
+    Awaitable,
+    Callable,
+    Dict,
+    List,
+    Optional,
+    Tuple,
+    Type,
+    Union,
+    overload,
+)
 
 from aws_lambda_powertools.middleware_factory import lambda_handler_decorator
 from aws_lambda_powertools.utilities.batch.exceptions import (
@@ -118,7 +129,7 @@ async def async_process():
             return list(await asyncio.gather(*[self._async_process_record(record) for record in self.records]))
 
         # WARNING
-        # Do not use "asyncio.run(async_process())" due to Lambda container thaws/freeze, otherwise we might get "Event Loop is closed"
+        # Do not use "asyncio.run(async_process())" due to Lambda container thaws/freeze, otherwise we might get "Event Loop is closed" # noqa: E501
         # Instead, get_event_loop() can also create one if a previous was erroneously closed
         # More: https://docs.aws.amazon.com/lambda/latest/dg/lambda-runtime-environment.html#runtimes-lifecycle-shutdown
         # Extra: just follow how the well-tested mangum library do:
@@ -127,7 +138,7 @@ async def async_process():
 
         # Detect environment and create a loop for each one
         coro = async_process()
-        if os.environ.get('AWS_LAMBDA_RUNTIME_API'):
+        if os.environ.get("AWS_LAMBDA_RUNTIME_API"):
             # Running in lambda server
             loop = asyncio.get_event_loop()
             task_instance = loop.create_task(coro)
@@ -393,7 +404,7 @@ def _clean(self):
         if self._entire_batch_failed():
             raise BatchProcessingError(
                 msg=f"All records failed processing. {len(self.exceptions)} individual errors logged "
-                    f"separately below.",
+                f"separately below.",
                 child_exceptions=self.exceptions,
             )
 
@@ -480,7 +491,7 @@ def _process_record(self, record: dict) -> Union[SuccessResponse, FailureRespons
 
 @lambda_handler_decorator
 def batch_processor(
-        handler: Callable, event: Dict, context: LambdaContext, record_handler: Callable, processor: BatchProcessor
+    handler: Callable, event: Dict, context: LambdaContext, record_handler: Callable, processor: BatchProcessor
 ):
     """
     Middleware to handle batch event processing
@@ -524,7 +535,6 @@ def batch_processor(
 
 
 class AsyncBatchProcessor(BasePartialBatchProcessor):
-
     def _process_record(self, record: dict):
         raise NotImplementedError()
 
@@ -551,7 +561,11 @@ async def _async_process_record(self, record: dict) -> Union[SuccessResponse, Fa
 
 @lambda_handler_decorator
 def async_batch_processor(
-        handler: Callable, event: Dict, context: LambdaContext, record_handler: Callable[..., Awaitable[Any]], processor: AsyncBatchProcessor
+    handler: Callable,
+    event: Dict,
+    context: LambdaContext,
+    record_handler: Callable[..., Awaitable[Any]],
+    processor: AsyncBatchProcessor,
 ):
     """
     Middleware to handle batch event processing

tests/functional/test_utilities_batch.py

@@ -1,16 +1,16 @@
 import json
 from random import randint
-from typing import Callable, Dict, Optional, Awaitable, Any
+from typing import Any, Awaitable, Callable, Dict, Optional
 
 import pytest
 from botocore.config import Config
 
 from aws_lambda_powertools.utilities.batch import (
-    BatchProcessor,
     AsyncBatchProcessor,
+    BatchProcessor,
     EventType,
-    batch_processor,
     async_batch_processor,
+    batch_processor,
 )
 from aws_lambda_powertools.utilities.batch.exceptions import BatchProcessingError
 from aws_lambda_powertools.utilities.data_classes.dynamo_db_stream_event import (