diff --git a/homeassistant/components/weatherflow_cloud/coordinator.py b/homeassistant/components/weatherflow_cloud/coordinator.py index 1b200f439b1e86..c1ccece20feb13 100644 --- a/homeassistant/components/weatherflow_cloud/coordinator.py +++ b/homeassistant/components/weatherflow_cloud/coordinator.py @@ -183,8 +183,10 @@ def _create_listen_message(self, device_id: int) -> RapidWindListenStartMessage: """Create rapid wind listen message.""" return RapidWindListenStartMessage(device_id=str(device_id)) - async def _handle_websocket_message(self, data: RapidWindWS) -> None: + async def _handle_websocket_message(self, data: RapidWindWS | None) -> None: """Handle rapid wind websocket data.""" + if data is None: + return device_id = data.device_id station_id = self.device_to_station_map[device_id] @@ -202,8 +204,12 @@ def _create_listen_message(self, device_id: int) -> ListenStartMessage: """Create observation listen message.""" return ListenStartMessage(device_id=str(device_id)) - async def _handle_websocket_message(self, data: ObservationTempestWS) -> None: + async def _handle_websocket_message( + self, data: ObservationTempestWS | None + ) -> None: """Handle observation websocket data.""" + if data is None: + return device_id = data.device_id station_id = self.device_to_station_map[device_id]