Fix state_class for sensors #127 #109

custom_components/localtuya/sensor.py

@@ -9,6 +9,7 @@
     DOMAIN,
     STATE_CLASSES_SCHEMA,
     SensorStateClass,
+    SensorEntity,
 )
 from homeassistant.const import (
     CONF_DEVICE_CLASS,
@@ -38,7 +39,7 @@ def flow_schema(dps):
     }
 
 
-class LocaltuyaSensor(LocalTuyaEntity):
+class LocaltuyaSensor(LocalTuyaEntity, SensorEntity):
     """Representation of a Tuya sensor."""
 
     def __init__(
@@ -53,7 +54,7 @@ def __init__(
         self._state = STATE_UNKNOWN
 
     @property
-    def state(self):
+    def native_value(self):
         """Return sensor state."""
         return self._state
 
@@ -68,7 +69,7 @@ def state_class(self) -> str | None:
         return self._config.get(CONF_STATE_CLASS)
 
     @property
-    def unit_of_measurement(self):
+    def native_unit_of_measurement(self):
         """Return the unit of measurement of this entity, if any."""
         return self._config.get(CONF_UNIT_OF_MEASUREMENT)