fix: addressed the rest of the comments

Author: rubenfonseca

docs/utilities/batch.md

@@ -313,7 +313,7 @@ For these scenarios, you can subclass `BatchProcessor` and quickly override `suc
 ???+ example
 	Let's suppose you'd like to add a metric named `BatchRecordFailures` for each batch record that failed processing
 
-```python title="Extending failure handling mechanism in BatchProcessor"
+```python hl_lines="8 9 16-19 22 38" title="Extending failure handling mechanism in BatchProcessor"
 --8<-- "examples/batch_processing/src/extending_failure.py"
 ```
 
@@ -328,7 +328,7 @@ You can create your own partial batch processor from scratch by inheriting the `
 
 You can then use this class as a context manager, or pass it to `batch_processor` to use as a decorator on your Lambda handler function.
 
-```python hl_lines="9 16 31 37 44 55 60 68" title="Creating a custom batch processor"
+```python hl_lines="9-12 20 35 41 48 59 64 68 76" title="Creating a custom batch processor"
 --8<-- "examples/batch_processing/src/custom_partial_processor.py"
 ```
 

examples/batch_processing/src/custom_partial_processor.py

@@ -6,14 +6,18 @@
 import boto3
 
 from aws_lambda_powertools import Logger
-from aws_lambda_powertools.utilities.batch import BasePartialProcessor, batch_processor
+from aws_lambda_powertools.utilities.batch import (
+    BasePartialBatchProcessor,
+    EventType,
+    process_partial_response,
+)
 
 table_name = os.getenv("TABLE_NAME", "table_not_found")
 
 logger = Logger()
 
 
-class MyPartialProcessor(BasePartialProcessor):
+class MyPartialProcessor(BasePartialBatchProcessor):
     """
     Process a record and stores successful results at a Amazon DynamoDB Table
 
@@ -26,7 +30,7 @@ class MyPartialProcessor(BasePartialProcessor):
     def __init__(self, table_name: str):
         self.table_name = table_name
 
-        super().__init__()
+        super().__init__(event_type=EventType.SQS)
 
     def _prepare(self):
         # It's called once, *before* processing
@@ -61,10 +65,12 @@ async def _async_process_record(self, record: dict):
         raise NotImplementedError()
 
 
+processor = MyPartialProcessor(table_name)
+
+
 def record_handler(record):
     return randint(0, 100)
 
 
-@batch_processor(record_handler=record_handler, processor=MyPartialProcessor(table_name))
 def lambda_handler(event, context):
-    return {"statusCode": 200}
+    return process_partial_response(event=event, record_handler=record_handler, processor=processor, context=context)

examples/batch_processing/src/disable_tracing.py

@@ -4,7 +4,7 @@
 from aws_lambda_powertools.utilities.batch import (
     BatchProcessor,
     EventType,
-    batch_processor,
+    process_partial_response,
 )
 from aws_lambda_powertools.utilities.data_classes.sqs_event import SQSRecord
 from aws_lambda_powertools.utilities.typing import LambdaContext
@@ -24,6 +24,5 @@ def record_handler(record: SQSRecord):
 
 @logger.inject_lambda_context
 @tracer.capture_lambda_handler
-@batch_processor(record_handler=record_handler, processor=processor)
 def lambda_handler(event, context: LambdaContext):
-    return processor.response()
+    return process_partial_response(event=event, record_handler=record_handler, processor=processor, context=context)