From 4ef04898e92fb78ca8b2f1b4b4fdce43f9b3d031 Mon Sep 17 00:00:00 2001 From: Robert Hillis Date: Tue, 17 Aug 2021 16:38:20 -0400 Subject: [PATCH] Fix goalzero sensor not using SensorEntity class (#54773) --- homeassistant/components/goalzero/sensor.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/homeassistant/components/goalzero/sensor.py b/homeassistant/components/goalzero/sensor.py index 31eadd55969..dbb85aa2d48 100644 --- a/homeassistant/components/goalzero/sensor.py +++ b/homeassistant/components/goalzero/sensor.py @@ -1,7 +1,11 @@ """Support for Goal Zero Yeti Sensors.""" from __future__ import annotations -from homeassistant.components.sensor import ATTR_LAST_RESET, ATTR_STATE_CLASS +from homeassistant.components.sensor import ( + ATTR_LAST_RESET, + ATTR_STATE_CLASS, + SensorEntity, +) from homeassistant.const import ( ATTR_DEVICE_CLASS, ATTR_NAME, @@ -36,7 +40,7 @@ async def async_setup_entry(hass, entry, async_add_entities): async_add_entities(sensors, True) -class YetiSensor(YetiEntity): +class YetiSensor(YetiEntity, SensorEntity): """Representation of a Goal Zero Yeti sensor.""" def __init__(self, api, coordinator, name, sensor_name, server_unique_id):