From d5f07ef45fd7597126812f4434b07a47dc985ca8 Mon Sep 17 00:00:00 2001 From: Marc Mueller <30130371+cdce8p@users.noreply.github.com> Date: Fri, 6 Oct 2023 13:44:07 +0200 Subject: [PATCH] Add override decorators to sensor (#94998) --- homeassistant/components/sensor/__init__.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/homeassistant/components/sensor/__init__.py b/homeassistant/components/sensor/__init__.py index 2cab631d1f0..55d2be6dfc6 100644 --- a/homeassistant/components/sensor/__init__.py +++ b/homeassistant/components/sensor/__init__.py @@ -11,6 +11,8 @@ import logging from math import ceil, floor, isfinite, log10 from typing import Any, Final, Self, cast, final +from typing_extensions import override + from homeassistant.config_entries import ConfigEntry # pylint: disable-next=hass-deprecated-import @@ -262,6 +264,7 @@ class SensorEntity(Entity): return self.device_class not in (None, SensorDeviceClass.ENUM) @property + @override def device_class(self) -> SensorDeviceClass | None: """Return the class of this entity.""" if hasattr(self, "_attr_device_class"): @@ -317,6 +320,7 @@ class SensorEntity(Entity): return None @property + @override def capability_attributes(self) -> Mapping[str, Any] | None: """Return the capability attributes.""" if state_class := self.state_class: @@ -362,6 +366,7 @@ class SensorEntity(Entity): @final @property + @override def state_attributes(self) -> dict[str, Any] | None: """Return state attributes.""" if last_reset := self.last_reset: @@ -439,6 +444,7 @@ class SensorEntity(Entity): @final @property + @override def unit_of_measurement(self) -> str | None: """Return the unit of measurement of the entity, after unit conversion.""" # Highest priority, for registered entities: unit set by user,with fallback to @@ -468,6 +474,7 @@ class SensorEntity(Entity): @final @property + @override def state(self) -> Any: """Return the state of the sensor and perform unit conversions, if needed.""" native_unit_of_measurement = self.native_unit_of_measurement