Measurement state class for pet recent weight and litter box litter w…

…eight
RobertD502 · Feb 23, 2023 · 15210da · 15210da
1 parent da01e3e
commit 15210da
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/custom_components/petkit/sensor.py b/custom_components/petkit/sensor.py
@@ -1570,7 +1570,7 @@ def entity_category(self) -> EntityCategory:
     def state_class(self) -> SensorStateClass:
         """Return the type of state class."""
 
-        return SensorStateClass.TOTAL
+        return SensorStateClass.MEASUREMENT
 
     @property
     def native_unit_of_measurement(self) -> UnitOfMass:
@@ -2241,7 +2241,7 @@ def device_class(self) -> SensorDeviceClass:
     def state_class(self) -> SensorStateClass:
         """Return the type of state class."""
 
-        return SensorStateClass.TOTAL
+        return SensorStateClass.MEASUREMENT
 
     def grab_recent_weight(self) -> float:
         """Grab the most recent weight."""