fix: mypy typing

heitorlessa · heitorlessa · commit 251541c79479 · 2021-12-13T14:32:49.000+01:00
diff --git a/aws_lambda_powertools/utilities/batch/base.py b/aws_lambda_powertools/utilities/batch/base.py
@@ -28,7 +28,9 @@
     from aws_lambda_powertools.utilities.parser.models import KinesisDataStreamRecord as KinesisDataStreamRecordModel
     from aws_lambda_powertools.utilities.parser.models import SqsRecordModel
 
-    BatchTypeModels = Union[SqsRecordModel, DynamoDBStreamRecordModel, KinesisDataStreamRecordModel]
+    BatchTypeModels = Optional[
+        Union[Type[SqsRecordModel], Type[DynamoDBStreamRecordModel], Type[KinesisDataStreamRecordModel]]
+    ]
 
 
 class EventType(Enum):
