diff --git a/homeassistant/components/rflink/__init__.py b/homeassistant/components/rflink/__init__.py index e703f0d09de..f5a0f0808da 100644 --- a/homeassistant/components/rflink/__init__.py +++ b/homeassistant/components/rflink/__init__.py @@ -6,7 +6,7 @@ from collections import defaultdict import logging import async_timeout -from rflink.protocol import create_rflink_connection +from rflink.protocol import ProtocolBase, create_rflink_connection from serial import SerialException import voluptuous as vol @@ -478,12 +478,16 @@ class RflinkCommand(RflinkDevice): # Keep repetition tasks to cancel if state is changed before repetitions # are sent - _repetition_task = None + _repetition_task: asyncio.Task[None] | None = None - _protocol = None + _protocol: ProtocolBase | None = None + + _wait_ack: bool | None = None @classmethod - def set_rflink_protocol(cls, protocol, wait_ack=None): + def set_rflink_protocol( + cls, protocol: ProtocolBase | None, wait_ack: bool | None = None + ) -> None: """Set the Rflink asyncio protocol as a class variable.""" cls._protocol = protocol if wait_ack is not None: