chickliu/ha_xiaomi_home
1
0
Fork 0
mirror of https://github.com/XiaoMi/ha_xiaomi_home.git synced 2026-01-15 22:10:43 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix: fix sensor entity type error

Browse Source
This commit is contained in:
topsworld 2025-01-17 11:54:06 +08:00
parent e7564b0152
commit 22f843d7ff
1 changed files with 4 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
custom_components/xiaomi_home/sensor.py
View File

@ -106,18 +106,15 @@ class Sensor(MIoTPropertyEntity, SensorEntity):
# device_class is not empty but unit is empty.
# Set the default unit according to device_class.
unit_sets = DEVICE_CLASS_UNITS.get(
self._attr_device_class, None)
self._attr_device_class, None) # type: ignore
self._attr_native_unit_of_measurement = list(
unit_sets)[0] if unit_sets else None
# Set state_class
if spec.state_class:
self._attr_state_class = spec.state_class
# Set icon
if spec.icon:
self._attr_icon = spec.icon
# Set state_class
if spec.state_class:
self._attr_state_class = spec.state_class
else:
# Set default state_class
self._attr_state_class = SensorStateClass.MEASUREMENT
@property
def native_value(self) -> Any: