From 91cf2e27923c0e40ba683e588306b95ab3696969 Mon Sep 17 00:00:00 2001 From: Franck Nijhof Date: Tue, 6 Dec 2022 14:50:19 +0100 Subject: [PATCH] Remove str from update device class (#83399) --- homeassistant/components/update/__init__.py | 6 +++--- pylint/plugins/hass_enforce_type_hints.py | 2 +- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/homeassistant/components/update/__init__.py b/homeassistant/components/update/__init__.py index 57e20a764ba..f8d00300d81 100644 --- a/homeassistant/components/update/__init__.py +++ b/homeassistant/components/update/__init__.py @@ -174,7 +174,7 @@ async def async_clear_skipped(entity: UpdateEntity, service_call: ServiceCall) - class UpdateEntityDescription(EntityDescription): """A class that describes update entities.""" - device_class: UpdateDeviceClass | str | None = None + device_class: UpdateDeviceClass | None = None entity_category: EntityCategory | None = EntityCategory.CONFIG @@ -184,7 +184,7 @@ class UpdateEntity(RestoreEntity): entity_description: UpdateEntityDescription _attr_auto_update: bool = False _attr_installed_version: str | None = None - _attr_device_class: UpdateDeviceClass | str | None + _attr_device_class: UpdateDeviceClass | None _attr_in_progress: bool | int = False _attr_latest_version: str | None = None _attr_release_summary: str | None = None @@ -206,7 +206,7 @@ class UpdateEntity(RestoreEntity): return self._attr_installed_version @property - def device_class(self) -> UpdateDeviceClass | str | None: + def device_class(self) -> UpdateDeviceClass | None: """Return the class of this entity.""" if hasattr(self, "_attr_device_class"): return self._attr_device_class diff --git a/pylint/plugins/hass_enforce_type_hints.py b/pylint/plugins/hass_enforce_type_hints.py index a60b1c7de85..1908a85ed9b 100644 --- a/pylint/plugins/hass_enforce_type_hints.py +++ b/pylint/plugins/hass_enforce_type_hints.py @@ -2325,7 +2325,7 @@ _INHERITANCE_MATCH: dict[str, list[ClassTypeHintMatch]] = { ), TypeHintMatch( function_name="device_class", - return_type=["UpdateDeviceClass", "str", None], + return_type=["UpdateDeviceClass", None], ), TypeHintMatch( function_name="in_progress",