home-assistant/core
1
0
Fork 1
mirror of https://github.com/home-assistant/core.git synced 2026-02-06 15:25:33 +01:00
Code Issues Packages Projects Releases Wiki Activity

Remove str from sensor state class (#161686)

Browse Source
This commit is contained in:
epenet
2026-02-06 12:01:23 +01:00
committed by GitHub
parent 47672614df
commit fbc91d3d3d
3 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
homeassistant/components/sensor/__init__.py
View File

@@ -112,7 +112,7 @@ class SensorEntityDescription(EntityDescription, frozen_or_thawed=True):
last_reset: datetime | None = None
native_unit_of_measurement: str | None = None
options: list[str] | None = None
state_class: SensorStateClass | str | None = None
state_class: SensorStateClass | None = None
suggested_display_precision: int | None = None
suggested_unit_of_measurement: str | None = None
unit_of_measurement: None = None # Type override, use native_unit_of_measurement
@@ -120,7 +120,7 @@ class SensorEntityDescription(EntityDescription, frozen_or_thawed=True):
def _numeric_state_expected(
device_class: SensorDeviceClass | None,
state_class: SensorStateClass | str | None,
state_class: SensorStateClass | None,
native_unit_of_measurement: str | None,
suggested_display_precision: int | None,
) -> bool:
@@ -188,7 +188,7 @@ class SensorEntity(Entity, cached_properties=CACHED_PROPERTIES_WITH_ATTR_):
_attr_native_unit_of_measurement: str | None
_attr_native_value: StateType | date | datetime | Decimal = None
_attr_options: list[str] | None
_attr_state_class: SensorStateClass | str | None
_attr_state_class: SensorStateClass | None
_attr_state: None = None # Subclasses of SensorEntity should not set this
_attr_suggested_display_precision: int | None
_attr_suggested_unit_of_measurement: str | None
@@ -329,7 +329,7 @@ class SensorEntity(Entity, cached_properties=CACHED_PROPERTIES_WITH_ATTR_):
return None
@cached_property
def state_class(self) -> SensorStateClass | str | None:
def state_class(self) -> SensorStateClass | None:
"""Return the state class of this entity, if any."""
if hasattr(self, "_attr_state_class"):
return self._attr_state_class

2
homeassistant/components/sensor/const.py
View File

@@ -835,7 +835,7 @@ DEVICE_CLASS_STATE_CLASSES: dict[SensorDeviceClass, set[SensorStateClass]] = {
}
STATE_CLASS_UNITS: dict[SensorStateClass | str, set[type[StrEnum] | str | None]] = {
STATE_CLASS_UNITS: dict[SensorStateClass, set[type[StrEnum] | str | None]] = {
SensorStateClass.MEASUREMENT_ANGLE: {DEGREE},
}

2
pylint/plugins/hass_enforce_type_hints.py
View File

@@ -2526,7 +2526,7 @@ _INHERITANCE_MATCH: dict[str, list[ClassTypeHintMatch]] = {
),
TypeHintMatch(
function_name="state_class",
return_type=["SensorStateClass", "str", None],
return_type=["SensorStateClass", None],
mandatory=True,
),
TypeHintMatch(