From 1990b5b6087353a3f5409381bd02fbd9440dde7a Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Daniel=20Hjelseth=20H=C3=B8yer?= Date: Wed, 8 Dec 2021 21:46:30 +0100 Subject: [PATCH] Use new EntityCategory enum in Surepetcare (#61282) --- homeassistant/components/surepetcare/binary_sensor.py | 6 +++--- homeassistant/components/surepetcare/sensor.py | 10 +++------- 2 files changed, 6 insertions(+), 10 deletions(-) diff --git a/homeassistant/components/surepetcare/binary_sensor.py b/homeassistant/components/surepetcare/binary_sensor.py index d53252fae9b..f4565ea3757 100644 --- a/homeassistant/components/surepetcare/binary_sensor.py +++ b/homeassistant/components/surepetcare/binary_sensor.py @@ -12,8 +12,8 @@ from homeassistant.components.binary_sensor import ( BinarySensorEntity, ) from homeassistant.config_entries import ConfigEntry -from homeassistant.const import ENTITY_CATEGORY_DIAGNOSTIC from homeassistant.core import HomeAssistant, callback +from homeassistant.helpers.entity import EntityCategory from homeassistant.helpers.entity_platform import AddEntitiesCallback from . import SurePetcareDataCoordinator @@ -67,7 +67,7 @@ class Hub(SurePetcareBinarySensor): """Sure Petcare Hub.""" _attr_device_class = BinarySensorDeviceClass.CONNECTIVITY - _attr_entity_category = ENTITY_CATEGORY_DIAGNOSTIC + _attr_entity_category = EntityCategory.DIAGNOSTIC @property def available(self) -> bool: @@ -117,7 +117,7 @@ class DeviceConnectivity(SurePetcareBinarySensor): """Sure Petcare Device.""" _attr_device_class = BinarySensorDeviceClass.CONNECTIVITY - _attr_entity_category = ENTITY_CATEGORY_DIAGNOSTIC + _attr_entity_category = EntityCategory.DIAGNOSTIC def __init__( self, diff --git a/homeassistant/components/surepetcare/sensor.py b/homeassistant/components/surepetcare/sensor.py index e53f319bdc5..e9967054900 100644 --- a/homeassistant/components/surepetcare/sensor.py +++ b/homeassistant/components/surepetcare/sensor.py @@ -9,13 +9,9 @@ from surepy.enums import EntityType from homeassistant.components.sensor import SensorDeviceClass, SensorEntity from homeassistant.config_entries import ConfigEntry -from homeassistant.const import ( - ATTR_VOLTAGE, - ENTITY_CATEGORY_DIAGNOSTIC, - PERCENTAGE, - VOLUME_MILLILITERS, -) +from homeassistant.const import ATTR_VOLTAGE, PERCENTAGE, VOLUME_MILLILITERS from homeassistant.core import HomeAssistant, callback +from homeassistant.helpers.entity import EntityCategory from homeassistant.helpers.entity_platform import AddEntitiesCallback from . import SurePetcareDataCoordinator @@ -52,7 +48,7 @@ class SureBattery(SurePetcareEntity, SensorEntity): """A sensor implementation for Sure Petcare batteries.""" _attr_device_class = SensorDeviceClass.BATTERY - _attr_entity_category = ENTITY_CATEGORY_DIAGNOSTIC + _attr_entity_category = EntityCategory.DIAGNOSTIC _attr_native_unit_of_measurement = PERCENTAGE def __init__(