io.Socket refactor

pylabrobot/io/io.py

@@ -3,11 +3,11 @@
 
 class IOBase(ABC):
   @abstractmethod
-  async def write(self, *args, **kwargs):
+  async def write(self, data: bytes, *args, **kwargs):
     pass
 
   @abstractmethod
-  async def read(self, *args, **kwargs):
+  async def read(self, *args, **kwargs) -> bytes:
     pass
 
   def serialize(self):