From 03b11501155e293418de6bd6c7282a405fe2a404 Mon Sep 17 00:00:00 2001 From: Aaron Bach Date: Wed, 20 Oct 2021 15:57:45 -0600 Subject: [PATCH] Add entity categories for appropriate Notion entities (#58105) --- homeassistant/components/notion/binary_sensor.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/homeassistant/components/notion/binary_sensor.py b/homeassistant/components/notion/binary_sensor.py index c064e492b16..ab3d0775436 100644 --- a/homeassistant/components/notion/binary_sensor.py +++ b/homeassistant/components/notion/binary_sensor.py @@ -16,6 +16,7 @@ from homeassistant.components.binary_sensor import ( BinarySensorEntityDescription, ) from homeassistant.config_entries import ConfigEntry +from homeassistant.const import ENTITY_CATEGORY_DIAGNOSTIC from homeassistant.core import HomeAssistant, callback from homeassistant.helpers.entity_platform import AddEntitiesCallback @@ -56,6 +57,7 @@ BINARY_SENSOR_DESCRIPTIONS = ( key=SENSOR_BATTERY, name="Low Battery", device_class=DEVICE_CLASS_BATTERY, + entity_category=ENTITY_CATEGORY_DIAGNOSTIC, on_state="critical", ), NotionBinarySensorDescription( @@ -80,6 +82,7 @@ BINARY_SENSOR_DESCRIPTIONS = ( key=SENSOR_MISSING, name="Missing", device_class=DEVICE_CLASS_CONNECTIVITY, + entity_category=ENTITY_CATEGORY_DIAGNOSTIC, on_state="not_missing", ), NotionBinarySensorDescription(