Skip to content

Commit 53b8e75

Browse files
committed
chore: address Tom's feedback on type name
1 parent 4a7ceea commit 53b8e75

File tree

1 file changed

+5
-5
lines changed
  • aws_lambda_powertools/utilities/batch

1 file changed

+5
-5
lines changed

Diff for: aws_lambda_powertools/utilities/batch/base.py

+5-5
Original file line numberDiff line numberDiff line change
@@ -48,8 +48,8 @@ class EventType(Enum):
4848
# When using Pydantic Models, it'll accept any
4949
EventSourceDataClassTypes = Union[SQSRecord, KinesisStreamRecord, DynamoDBRecord]
5050
BatchEventTypes = Union[EventSourceDataClassTypes, "BatchTypeModels"]
51-
SuccessCallback = Tuple[str, Any, BatchEventTypes]
52-
FailureCallback = Tuple[str, str, BatchEventTypes]
51+
SuccessResponse = Tuple[str, Any, BatchEventTypes]
52+
FailureResponse = Tuple[str, str, BatchEventTypes]
5353

5454

5555
class BasePartialProcessor(ABC):
@@ -111,7 +111,7 @@ def __call__(self, records: List[dict], handler: Callable):
111111
self.handler = handler
112112
return self
113113

114-
def success_handler(self, record, result: Any) -> SuccessCallback:
114+
def success_handler(self, record, result: Any) -> SuccessResponse:
115115
"""
116116
Success callback
117117
@@ -124,7 +124,7 @@ def success_handler(self, record, result: Any) -> SuccessCallback:
124124
self.success_messages.append(record)
125125
return entry
126126

127-
def failure_handler(self, record, exception: _OptExcInfo) -> FailureCallback:
127+
def failure_handler(self, record, exception: _OptExcInfo) -> FailureResponse:
128128
"""
129129
Failure callback
130130
@@ -228,7 +228,7 @@ def _prepare(self):
228228
self.fail_messages.clear()
229229
self.batch_response = self.DEFAULT_RESPONSE
230230

231-
def _process_record(self, record: dict) -> Union[SuccessCallback, FailureCallback]:
231+
def _process_record(self, record: dict) -> Union[SuccessResponse, FailureResponse]:
232232
"""
233233
Process a record with instance's handler
234234

0 commit comments

Comments
 (0)