From e5f0b450cf54175d7be6c3ddf41bcb087ef2d7b1 Mon Sep 17 00:00:00 2001
From: Tyler Goodlet <jgbt@protonmail.com>
Date: Thu, 18 Apr 2024 15:40:26 -0400
Subject: [PATCH] Add some `bytes` annots

---
 tractor/_ipc.py | 6 ++----
 1 file changed, 2 insertions(+), 4 deletions(-)

diff --git a/tractor/_ipc.py b/tractor/_ipc.py
index a5b44a4..f76d4ef 100644
--- a/tractor/_ipc.py
+++ b/tractor/_ipc.py
@@ -314,8 +314,7 @@ class MsgpackTCPStream(MsgTransport):
 
         while True:
             try:
-                header = await self.recv_stream.receive_exactly(4)
-
+                header: bytes = await self.recv_stream.receive_exactly(4)
             except (
                 ValueError,
                 ConnectionResetError,
@@ -337,8 +336,7 @@ class MsgpackTCPStream(MsgTransport):
             size, = struct.unpack("<I", header)
 
             log.transport(f'received header {size}')  # type: ignore
-
-            msg_bytes = await self.recv_stream.receive_exactly(size)
+            msg_bytes: bytes = await self.recv_stream.receive_exactly(size)
 
             log.transport(f"received {msg_bytes}")  # type: ignore
             try: