Remove str from update device class (#83399)

This commit is contained in:
Franck Nijhof 2022-12-06 14:50:19 +01:00 committed by GitHub
parent 923fa473e1
commit 91cf2e2792
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 4 additions and 4 deletions

View file

@ -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

View file

@ -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",