From 9c3aa8156d94691555c30684fa4fc2b88e0b683a Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Daniel=20Hjelseth=20H=C3=B8yer?= Date: Sun, 17 Oct 2021 19:26:29 +0200 Subject: [PATCH] Add category diagnostic to Surepetcare battery sensor (#57852) --- homeassistant/components/surepetcare/sensor.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/homeassistant/components/surepetcare/sensor.py b/homeassistant/components/surepetcare/sensor.py index c0441d0a7fd..cc4fe01fffa 100644 --- a/homeassistant/components/surepetcare/sensor.py +++ b/homeassistant/components/surepetcare/sensor.py @@ -12,6 +12,7 @@ from homeassistant.config_entries import ConfigEntry from homeassistant.const import ( ATTR_VOLTAGE, DEVICE_CLASS_BATTERY, + ENTITY_CATEGORY_DIAGNOSTIC, PERCENTAGE, VOLUME_MILLILITERS, ) @@ -49,9 +50,10 @@ async def async_setup_entry( class SureBattery(SurePetcareEntity, SensorEntity): - """A sensor implementation for Sure Petcare Entities.""" + """A sensor implementation for Sure Petcare batteries.""" _attr_device_class = DEVICE_CLASS_BATTERY + _attr_entity_category = ENTITY_CATEGORY_DIAGNOSTIC _attr_native_unit_of_measurement = PERCENTAGE def __init__(