From 490229d016e6bfe127a05bd7df3ec3778d6e9ee8 Mon Sep 17 00:00:00 2001 From: David Brochart Date: Mon, 18 Nov 2024 16:32:45 +0100 Subject: [PATCH] Types --- .../jupyter-server-ydoc/jupyter_server_ydoc/test_utils.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/projects/jupyter-server-ydoc/jupyter_server_ydoc/test_utils.py b/projects/jupyter-server-ydoc/jupyter_server_ydoc/test_utils.py index 8ced6ea0..24528004 100644 --- a/projects/jupyter-server-ydoc/jupyter_server_ydoc/test_utils.py +++ b/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)