Fix log warnings

masaccio · Mar 4, 2023 · abf7e25 · abf7e25
1 parent c4e9e33
commit abf7e25
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/custom_components/kingspan_watchman_sensit/sensor.py b/custom_components/kingspan_watchman_sensit/sensor.py
@@ -66,7 +66,6 @@ def icon(self):
 
 class TankPercentageFull(SENSiTEntity, SensorEntity):
     _attr_name = "Tank Percentage Full"
-    _attr_device_class = SensorDeviceClass.VOLUME
     _attr_native_unit_of_measurement = PERCENTAGE
     _attr_state_class = SensorStateClass.TOTAL
 
@@ -94,7 +93,6 @@ class TankCapacity(SENSiTEntity, SensorEntity):
     _attr_name = "Tank Capacity"
     _attr_device_class = SensorDeviceClass.VOLUME
     _attr_native_unit_of_measurement = UnitOfVolume.LITERS
-    _attr_state_class = SensorStateClass.MEASUREMENT
 
     @property
     def native_value(self):
@@ -125,7 +123,7 @@ class CurrentUsage(SENSiTEntity, SensorEntity):
     _attr_name = "Current Usage"
     _attr_device_class = SensorDeviceClass.VOLUME
     _attr_native_unit_of_measurement = UnitOfVolume.LITERS
-    _attr_state_class = SensorStateClass.MEASUREMENT
+    _attr_state_class = SensorStateClass.TOTAL
 
     @property
     def native_value(self):