add missing return types

Author: lev-blit

stubs/simple-websocket/simple_websocket/aiows.pyi

@@ -41,7 +41,7 @@ class AioBase:
     async def connect(self) -> None: ...
     async def handshake(self) -> None: ...
     async def send(self, data: bytes | Any) -> None: ...
-    async def receive(self, timeout: float | None = None): ...
+    async def receive(self, timeout: float | None = None) -> bytes | str | Any: ...
     async def close(self, reason: CloseReason | None = None, message: str | None = None) -> None: ...
     def choose_subprotocol(self, request: Request) -> str | None: ...
 

stubs/simple-websocket/simple_websocket/ws.pyi

@@ -21,7 +21,7 @@ class _ThreadClassProtocol(Protocol):
 class _EventClassProtocol(Protocol):
     def clear(self) -> None: ...
     def set(self) -> None: ...
-    def wait(self, timeout: float | None = None): ...
+    def wait(self, timeout: float | None = None) -> bool: ...
 
 @type_check_only
 class _SelectorClassProtocol(Protocol):
@@ -127,6 +127,6 @@ class Client(Base):
         ssl_context: SSLContext | None = None,
         thread_class: type[_ThreadClassProtocol] | None = None,
         event_class: type[_EventClassProtocol] | None = None,
-    ): ...
+    ) -> Client: ...
     def handshake(self) -> None: ...
     def close(self, reason: CloseReason | None = None, message: str | None = None) -> None: ...