diff --git a/custom_components/ecowitt/sensor.py b/custom_components/ecowitt/sensor.py index 6dabcc8..492c92e 100644 --- a/custom_components/ecowitt/sensor.py +++ b/custom_components/ecowitt/sensor.py @@ -4,6 +4,7 @@ from . import EcowittEntity, async_add_ecowitt_entities from homeassistant.components.sensor import DOMAIN as SENSOR_DOMAIN +from homeassistant.components.sensor import SensorEntity from homeassistant.helpers.dispatcher import async_dispatcher_connect from .const import ( DOMAIN, @@ -35,7 +36,7 @@ def add_entities(discovery_info=None): add_entities(hass.data[DOMAIN][entry.entry_id][REG_ENTITIES][TYPE_SENSOR]) -class EcowittSensor(EcowittEntity): +class EcowittSensor(EcowittEntity, SensorEntity): def __init__(self, hass, entry, key, name, dc, uom, icon, sc): """Initialize the sensor."""