🐛 fix for DrinkTimes sensors

Jezza34000 · Dec 22, 2024 · 3fba83d · 3fba83d
1 parent 3a1ca64
commit 3fba83d
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/custom_components/petkit/sensor.py b/custom_components/petkit/sensor.py
@@ -449,7 +449,11 @@ class PetKitSensorDesc(PetKitDescSensorBase, SensorEntityDescription):
             translation_key="drink_times",
             entity_category=EntityCategory.DIAGNOSTIC,
             state_class=SensorStateClass.MEASUREMENT,
-            value=lambda device: (len(device.device_records)),
+            value=lambda device: (
+                len(device.device_records)
+                if isinstance(device.device_records, list)
+                else None
+            ),
         ),
     ],
     Pet: [