From 078c3ff8482bd272e53a8f6030517eea73a8edee Mon Sep 17 00:00:00 2001
From: "J. Nick Koston" <nick@koston.org>
Date: Sun, 19 May 2024 21:32:50 -1000
Subject: [PATCH] Fix missing type for mqtt websocket wrapper (#117752)

---
 homeassistant/components/mqtt/client.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/homeassistant/components/mqtt/client.py b/homeassistant/components/mqtt/client.py
index 80667f812e03d..830ab5380967b 100644
--- a/homeassistant/components/mqtt/client.py
+++ b/homeassistant/components/mqtt/client.py
@@ -99,7 +99,7 @@
 TIMEOUT_ACK = 10
 RECONNECT_INTERVAL_SECONDS = 10
 
-type SocketType = socket.socket | ssl.SSLSocket | Any
+type SocketType = socket.socket | ssl.SSLSocket | mqtt.WebsocketWrapper | Any
 
 type SubscribePayloadType = str | bytes  # Only bytes if encoding is None