From b56eabc35bd59316276ea31daabddb21635ff140 Mon Sep 17 00:00:00 2001 From: epenet <6771947+epenet@users.noreply.github.com> Date: Thu, 15 Sep 2022 11:58:24 +0200 Subject: [PATCH] Enable disallow-any-generics in number (#78502) --- homeassistant/components/number/__init__.py | 12 ++++++++---- 1 file changed, 8 insertions(+), 4 deletions(-) diff --git a/homeassistant/components/number/__init__.py b/homeassistant/components/number/__init__.py index 4990fbfa7f8..5f92bdfed8b 100644 --- a/homeassistant/components/number/__init__.py +++ b/homeassistant/components/number/__init__.py @@ -78,10 +78,12 @@ VALID_UNITS: dict[str, tuple[str, ...]] = { NumberDeviceClass.TEMPERATURE: temperature_util.VALID_UNITS, } +# mypy: disallow-any-generics + async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool: """Set up Number entities.""" - component = hass.data[DOMAIN] = EntityComponent( + component = hass.data[DOMAIN] = EntityComponent[NumberEntity]( _LOGGER, DOMAIN, hass, SCAN_INTERVAL ) await component.async_setup(config) @@ -115,13 +117,13 @@ async def async_set_value(entity: NumberEntity, service_call: ServiceCall) -> No async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: """Set up a config entry.""" - component: EntityComponent = hass.data[DOMAIN] + component: EntityComponent[NumberEntity] = hass.data[DOMAIN] return await component.async_setup_entry(entry) async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: """Unload a config entry.""" - component: EntityComponent = hass.data[DOMAIN] + component: EntityComponent[NumberEntity] = hass.data[DOMAIN] return await component.async_unload_entry(entry) @@ -423,7 +425,9 @@ class NumberEntity(Entity): """Set new value.""" await self.hass.async_add_executor_job(self.set_value, value) - def _convert_to_state_value(self, value: float, method: Callable) -> float: + def _convert_to_state_value( + self, value: float, method: Callable[[float, int], float] + ) -> float: """Convert a value in the number's native unit to the configured unit.""" native_unit_of_measurement = self.native_unit_of_measurement