Fix native and suggested units for energy

dan-r · Dec 28, 2023 · 16aadf4 · 16aadf4
1 parent 8ad87f1
commit 16aadf4
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/custom_components/ohme/sensor.py b/custom_components/ohme/sensor.py
@@ -76,7 +76,9 @@ def native_value(self):
 class EnergyUsageSensor(CoordinatorEntity[OhmeStatisticsCoordinator], SensorEntity):
     """Sensor for total energy usage."""
     _attr_name = "Accumulative Energy Usage"
-    _attr_native_unit_of_measurement = UnitOfEnergy.KILO_WATT_HOUR
+    _attr_native_unit_of_measurement = UnitOfEnergy.WATT_HOUR
+    _attr_suggested_unit_of_measurement = UnitOfEnergy.KILO_WATT_HOUR
+    _attr_suggested_display_precision = 1
     _attr_device_class = SensorDeviceClass.ENERGY
 
     def __init__(
@@ -111,7 +113,7 @@ def icon(self):
     def native_value(self):
         """Get value from data returned from API by coordinator"""
         if self.coordinator.data and self.coordinator.data['energyChargedTotalWh']:
-            return self.coordinator.data['energyChargedTotalWh'] / 1000
+            return self.coordinator.data['energyChargedTotalWh']
 
         return None