Move state_class to class attribute.

djtimca · Feb 28, 2023 · 89ff16b · 89ff16b
1 parent 69d4a81
commit 89ff16b
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/custom_components/googlewifi/sensor.py b/custom_components/googlewifi/sensor.py
@@ -148,6 +148,8 @@ def __init__(
         self.attrs = {}
         self._unit_of_measurement = unit_of_measure
 
+    _attr_state_class = SensorStateClass.MEASUREMENT
+
     @property
     def unique_id(self):
         """Return the unique id for this sensor."""
@@ -178,11 +180,6 @@ def state(self):
                     self._state = unit_convert(self._state, self._unit_of_measurement)
 
             return self._state
-
-    @property
-    def state_class(self):
-        """Return the state class of total for the sensors."""
-        return SensorStateClass.MEASUREMENT
 
     @property
     def unit_of_measurement(self):