From 34b936da9855e019b75fec345ab8dc82d795dbe3 Mon Sep 17 00:00:00 2001
From: Simone Chemelli <simone.chemelli@gmail.com>
Date: Mon, 25 Oct 2021 09:44:15 +0200
Subject: [PATCH] Add entity category to Neato (#58367)

---
 homeassistant/components/neato/sensor.py | 7 ++++++-
 homeassistant/components/neato/switch.py | 7 ++++++-
 2 files changed, 12 insertions(+), 2 deletions(-)

diff --git a/homeassistant/components/neato/sensor.py b/homeassistant/components/neato/sensor.py
index 9a5da4c1950..1fd6547d926 100644
--- a/homeassistant/components/neato/sensor.py
+++ b/homeassistant/components/neato/sensor.py
@@ -11,7 +11,7 @@ from pybotvac.robot import Robot
 from homeassistant.components.neato import NeatoHub
 from homeassistant.components.sensor import DEVICE_CLASS_BATTERY, SensorEntity
 from homeassistant.config_entries import ConfigEntry
-from homeassistant.const import PERCENTAGE
+from homeassistant.const import ENTITY_CATEGORY_DIAGNOSTIC, PERCENTAGE
 from homeassistant.core import HomeAssistant
 from homeassistant.helpers.entity import DeviceInfo
 from homeassistant.helpers.entity_platform import AddEntitiesCallback
@@ -83,6 +83,11 @@ class NeatoSensor(SensorEntity):
         """Return the device class."""
         return DEVICE_CLASS_BATTERY
 
+    @property
+    def entity_category(self) -> str:
+        """Device entity category."""
+        return ENTITY_CATEGORY_DIAGNOSTIC
+
     @property
     def available(self) -> bool:
         """Return availability."""
diff --git a/homeassistant/components/neato/switch.py b/homeassistant/components/neato/switch.py
index 65d296bcb28..b6a1302bda7 100644
--- a/homeassistant/components/neato/switch.py
+++ b/homeassistant/components/neato/switch.py
@@ -10,7 +10,7 @@ from pybotvac.robot import Robot
 
 from homeassistant.components.neato import NeatoHub
 from homeassistant.config_entries import ConfigEntry
-from homeassistant.const import STATE_OFF, STATE_ON
+from homeassistant.const import ENTITY_CATEGORY_CONFIG, STATE_OFF, STATE_ON
 from homeassistant.core import HomeAssistant
 from homeassistant.helpers.entity import DeviceInfo, ToggleEntity
 from homeassistant.helpers.entity_platform import AddEntitiesCallback
@@ -106,6 +106,11 @@ class NeatoConnectedSwitch(ToggleEntity):
             self.type == SWITCH_TYPE_SCHEDULE and self._schedule_state == STATE_ON
         )
 
+    @property
+    def entity_category(self) -> str:
+        """Device entity category."""
+        return ENTITY_CATEGORY_CONFIG
+
     @property
     def device_info(self) -> DeviceInfo:
         """Device info for neato robot."""