Types

projects/jupyter-server-ydoc/jupyter_server_ydoc/test_utils.py

@@ -59,7 +59,7 @@ def emit(self, schema_id: str, data: dict) -> None:
 
 
 class Websocket:
-    def __init__(self, websocket, path: str):
+    def __init__(self, websocket: Any, path: str):
         self._websocket = websocket
         self._path = path
         self._send_lock = Lock()
@@ -78,7 +78,7 @@ async def __anext__(self) -> bytes:
             raise StopAsyncIteration()
         return message
 
-    async def send(self, message: bytes):
+    async def send(self, message: bytes) -> None:
         async with self._send_lock:
             await self._websocket.send_bytes(message)