Clean up typing

Author: nateprewitt

packages/aws-event-stream/src/aws_event_stream/_private/serializers.py

@@ -70,11 +70,13 @@ async def send(self, event: E) -> None:
             )
         if self._signer is not None:
             encoder = self._serializer.event_header_encoder_cls
-            result = await self._signer.sign_event(
+            result = await self._signer.sign_event(  # type: ignore
                 event_message=result,
                 event_encoder_cls=encoder,
             )
-        encoded_result = result.encode()
+
+        assert isinstance(result, EventMessage)
+        encoded_result: bytes = result.encode()
         try:
             await self._writer.write(encoded_result)
         except Exception as e:

packages/aws-event-stream/src/aws_event_stream/events.py

@@ -387,7 +387,7 @@ def get_result(self) -> bytes:
             raise InvalidHeadersLength(len(result))
         return result
 
-    def encode_headers(self, headers: HEADERS_DICT):
+    def encode_headers(self, headers: HEADERS_DICT) -> None:
         """Encode a map of headers.
 
         :param headers: A mapping of headers to encode.

packages/aws-sdk-signers/src/aws_sdk_signers/signers.py

@@ -833,7 +833,7 @@ async def sign_event(
                 )
 
             timestamp = new_signing_properties["date"]
-            headers: dict[str, str | bytes] = {":date": date_obj}
+            headers: dict[str, str | bytes | datetime.datetime] = {":date": date_obj}
             encoder = event_encoder_cls()
             encoder.encode_headers(headers)
             encoded_headers = encoder.get_result()