From 8cb703526d3d8168f72da90be0e7023856422183 Mon Sep 17 00:00:00 2001 From: Yordan Suarez Date: Wed, 29 Dec 2021 14:27:25 -0500 Subject: [PATCH] replace FplEntity with FplEnergyEntity --- custom_components/fpl/sensor_KWHSensor.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/custom_components/fpl/sensor_KWHSensor.py b/custom_components/fpl/sensor_KWHSensor.py index 71abae2..6d48d03 100644 --- a/custom_components/fpl/sensor_KWHSensor.py +++ b/custom_components/fpl/sensor_KWHSensor.py @@ -32,7 +32,8 @@ class BillToDateKWHSensor(FplEnergyEntity): def state_class(self) -> str: """Return the state class of this entity, from STATE_CLASSES, if any.""" -class NetReceivedKWHSensor(FplEntity): + +class NetReceivedKWHSensor(FplEnergyEntity): def __init__(self, coordinator, config, account): super().__init__(coordinator, config, account, "Received Meter Reading KWH") @@ -44,7 +45,8 @@ class NetReceivedKWHSensor(FplEntity): def icon(self): return "mdi:flash" -class NetDeliveredKWHSensor(FplEntity): + +class NetDeliveredKWHSensor(FplEnergyEntity): def __init__(self, coordinator, config, account): super().__init__(coordinator, config, account, "Delivered Meter Reading KWH")