Controller battery category changed to diagnostic

sanjoyg · May 25, 2024 · 0e912da · 0e912da
1 parent 43219ee
commit 0e912da
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/custom_components/dirigera_platform/sensor.py b/custom_components/dirigera_platform/sensor.py
@@ -11,6 +11,7 @@
 from homeassistant.components.sensor import SensorDeviceClass, SensorEntity
 from homeassistant.const import CONF_IP_ADDRESS, CONF_TOKEN
 from homeassistant.core import HomeAssistantError
+from homeassistant.helpers.entity import EntityCategory
 
 from .const import DOMAIN
 from .base_classes import ikea_base_device, ikea_base_device_sensor
@@ -204,6 +205,10 @@ def __init__(self,hass:core.HomeAssistant, hub:Hub, json_data:Controller):
         logger.debug("ikea_controller ctor...")
         super().__init__(hass , hub, json_data, hub.get_controller_by_id)
 
+    @property
+    def entity_category(self):
+        return EntityCategory.DIAGNOSTIC
+
     @property
     def icon(self):
         return "mdi:battery"