Remove str from update device class (#83399)
This commit is contained in:
parent
923fa473e1
commit
91cf2e2792
2 changed files with 4 additions and 4 deletions
|
@ -174,7 +174,7 @@ async def async_clear_skipped(entity: UpdateEntity, service_call: ServiceCall) -
|
||||||
class UpdateEntityDescription(EntityDescription):
|
class UpdateEntityDescription(EntityDescription):
|
||||||
"""A class that describes update entities."""
|
"""A class that describes update entities."""
|
||||||
|
|
||||||
device_class: UpdateDeviceClass | str | None = None
|
device_class: UpdateDeviceClass | None = None
|
||||||
entity_category: EntityCategory | None = EntityCategory.CONFIG
|
entity_category: EntityCategory | None = EntityCategory.CONFIG
|
||||||
|
|
||||||
|
|
||||||
|
@ -184,7 +184,7 @@ class UpdateEntity(RestoreEntity):
|
||||||
entity_description: UpdateEntityDescription
|
entity_description: UpdateEntityDescription
|
||||||
_attr_auto_update: bool = False
|
_attr_auto_update: bool = False
|
||||||
_attr_installed_version: str | None = None
|
_attr_installed_version: str | None = None
|
||||||
_attr_device_class: UpdateDeviceClass | str | None
|
_attr_device_class: UpdateDeviceClass | None
|
||||||
_attr_in_progress: bool | int = False
|
_attr_in_progress: bool | int = False
|
||||||
_attr_latest_version: str | None = None
|
_attr_latest_version: str | None = None
|
||||||
_attr_release_summary: str | None = None
|
_attr_release_summary: str | None = None
|
||||||
|
@ -206,7 +206,7 @@ class UpdateEntity(RestoreEntity):
|
||||||
return self._attr_installed_version
|
return self._attr_installed_version
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def device_class(self) -> UpdateDeviceClass | str | None:
|
def device_class(self) -> UpdateDeviceClass | None:
|
||||||
"""Return the class of this entity."""
|
"""Return the class of this entity."""
|
||||||
if hasattr(self, "_attr_device_class"):
|
if hasattr(self, "_attr_device_class"):
|
||||||
return self._attr_device_class
|
return self._attr_device_class
|
||||||
|
|
|
@ -2325,7 +2325,7 @@ _INHERITANCE_MATCH: dict[str, list[ClassTypeHintMatch]] = {
|
||||||
),
|
),
|
||||||
TypeHintMatch(
|
TypeHintMatch(
|
||||||
function_name="device_class",
|
function_name="device_class",
|
||||||
return_type=["UpdateDeviceClass", "str", None],
|
return_type=["UpdateDeviceClass", None],
|
||||||
),
|
),
|
||||||
TypeHintMatch(
|
TypeHintMatch(
|
||||||
function_name="in_progress",
|
function_name="in_progress",
|
||||||
|
|
Loading…
Add table
Reference in a new issue