diff --git a/homeassistant/components/modbus/fan.py b/homeassistant/components/modbus/fan.py index cf5c9762db8..349ae0d0619 100644 --- a/homeassistant/components/modbus/fan.py +++ b/homeassistant/components/modbus/fan.py @@ -2,11 +2,13 @@ from __future__ import annotations import logging +from typing import Any from homeassistant.components.fan import FanEntity from homeassistant.const import CONF_NAME from homeassistant.core import HomeAssistant -from homeassistant.helpers.typing import ConfigType +from homeassistant.helpers.entity_platform import AddEntitiesCallback +from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType from . import get_hub from .base_platform import BaseSwitch @@ -18,8 +20,11 @@ _LOGGER = logging.getLogger(__name__) async def async_setup_platform( - hass: HomeAssistant, config: ConfigType, async_add_entities, discovery_info=None -): + hass: HomeAssistant, + config: ConfigType, + async_add_entities: AddEntitiesCallback, + discovery_info: DiscoveryInfoType | None = None, +) -> None: """Read configuration and create Modbus fans.""" if discovery_info is None: # pragma: no cover return @@ -39,13 +44,13 @@ class ModbusFan(BaseSwitch, FanEntity): speed: str | None = None, percentage: int | None = None, preset_mode: str | None = None, - **kwargs, + **kwargs: Any, ) -> None: """Set fan on.""" await self.async_turn(self.command_on) @property - def is_on(self): + def is_on(self) -> bool: """Return true if fan is on. This is needed due to the ongoing conversion of fan. diff --git a/homeassistant/components/modbus/light.py b/homeassistant/components/modbus/light.py index dd9a8ad754d..f0f2541ad0f 100644 --- a/homeassistant/components/modbus/light.py +++ b/homeassistant/components/modbus/light.py @@ -2,11 +2,13 @@ from __future__ import annotations import logging +from typing import Any from homeassistant.components.light import LightEntity from homeassistant.const import CONF_LIGHTS, CONF_NAME from homeassistant.core import HomeAssistant -from homeassistant.helpers.typing import ConfigType +from homeassistant.helpers.entity_platform import AddEntitiesCallback +from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType from . import get_hub from .base_platform import BaseSwitch @@ -17,8 +19,11 @@ _LOGGER = logging.getLogger(__name__) async def async_setup_platform( - hass: HomeAssistant, config: ConfigType, async_add_entities, discovery_info=None -): + hass: HomeAssistant, + config: ConfigType, + async_add_entities: AddEntitiesCallback, + discovery_info: DiscoveryInfoType | None = None, +) -> None: """Read configuration and create Modbus lights.""" if discovery_info is None: # pragma: no cover return @@ -33,6 +38,6 @@ async def async_setup_platform( class ModbusLight(BaseSwitch, LightEntity): """Class representing a Modbus light.""" - async def async_turn_on(self, **kwargs): + async def async_turn_on(self, **kwargs: Any) -> None: """Set light on.""" await self.async_turn(self.command_on) diff --git a/homeassistant/components/modbus/switch.py b/homeassistant/components/modbus/switch.py index 55dc014420f..86cba7c36ff 100644 --- a/homeassistant/components/modbus/switch.py +++ b/homeassistant/components/modbus/switch.py @@ -2,11 +2,13 @@ from __future__ import annotations import logging +from typing import Any from homeassistant.components.switch import SwitchEntity from homeassistant.const import CONF_NAME, CONF_SWITCHES from homeassistant.core import HomeAssistant -from homeassistant.helpers.typing import ConfigType +from homeassistant.helpers.entity_platform import AddEntitiesCallback +from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType from . import get_hub from .base_platform import BaseSwitch @@ -17,8 +19,11 @@ _LOGGER = logging.getLogger(__name__) async def async_setup_platform( - hass: HomeAssistant, config: ConfigType, async_add_entities, discovery_info=None -): + hass: HomeAssistant, + config: ConfigType, + async_add_entities: AddEntitiesCallback, + discovery_info: DiscoveryInfoType | None = None, +) -> None: """Read configuration and create Modbus switches.""" switches = [] @@ -34,6 +39,6 @@ async def async_setup_platform( class ModbusSwitch(BaseSwitch, SwitchEntity): """Base class representing a Modbus switch.""" - async def async_turn_on(self, **kwargs): + async def async_turn_on(self, **kwargs: Any) -> None: """Set switch on.""" await self.async_turn(self.command_on)