Fix type of on_ parameters (websocket-client#1006)

jpshivakavi · Nov 5, 2024 · 52329c4 · 52329c4
1 parent 8caa272
commit 52329c4
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/websocket/_app.py b/websocket/_app.py
@@ -55,11 +55,11 @@ def __init__(
         self,
         url: str,
         header: Union[list, dict, Callable, None] = None,
-        on_open: Optional[Callable[[WebSocket], None]] = None,
-        on_reconnect: Optional[Callable[[WebSocket], None]] = None,
-        on_message: Optional[Callable[[WebSocket, Any], None]] = None,
-        on_error: Optional[Callable[[WebSocket, Any], None]] = None,
-        on_close: Optional[Callable[[WebSocket, Any, Any], None]] = None,
+        on_open: Optional[Callable[["WebSocketApp"], None]] = None,
+        on_reconnect: Optional[Callable[["WebSocketApp"], None]] = None,
+        on_message: Optional[Callable[["WebSocketApp", Any], None]] = None,
+        on_error: Optional[Callable[["WebSocketApp", Any], None]] = None,
+        on_close: Optional[Callable[["WebSocketApp", Any, Any], None]] = None,
         on_ping: Optional[Callable] = None,
         on_pong: Optional[Callable] = None,
         on_cont_message: Optional[Callable] = None,