diff --git a/src/judas_client/connector.py b/src/judas_client/connector.py index 8f47e31..a68b64a 100644 --- a/src/judas_client/connector.py +++ b/src/judas_client/connector.py @@ -173,7 +173,7 @@ class Connector: self.inbound_buffer.split(b"\n", 1) ) try: - message = Message.from_bytes(message_bytes) + message: Message = Message.from_bytes(message_bytes) # handle incoming ACKs if ( message.category == Category.CONTROL @@ -192,7 +192,7 @@ class Connector: self.on_message(message) if message.ack_required: - ack_message = Message.ack(message.id) + ack_message: Message = Message.ack(message.id) self.send(ack_message) self._send_outbound() except Exception as e: