From ad778f53f732cae72e1081e73289acbcf570a20d Mon Sep 17 00:00:00 2001 From: Robert Hillis Date: Thu, 16 Dec 2021 06:34:01 -0500 Subject: [PATCH] Use enums in tractive (#62024) --- homeassistant/components/tractive/switch.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/homeassistant/components/tractive/switch.py b/homeassistant/components/tractive/switch.py index e606b68779e..2a425a8f2ac 100644 --- a/homeassistant/components/tractive/switch.py +++ b/homeassistant/components/tractive/switch.py @@ -9,9 +9,9 @@ from aiotractive.exceptions import TractiveError from homeassistant.components.switch import SwitchEntity, SwitchEntityDescription from homeassistant.config_entries import ConfigEntry -from homeassistant.const import ENTITY_CATEGORY_CONFIG from homeassistant.core import HomeAssistant, callback from homeassistant.helpers.dispatcher import async_dispatcher_connect +from homeassistant.helpers.entity import EntityCategory from homeassistant.helpers.entity_platform import AddEntitiesCallback from . import Trackables @@ -50,21 +50,21 @@ SWITCH_TYPES: tuple[TractiveSwitchEntityDescription, ...] = ( name="Tracker Buzzer", icon="mdi:volume-high", method="async_set_buzzer", - entity_category=ENTITY_CATEGORY_CONFIG, + entity_category=EntityCategory.CONFIG, ), TractiveSwitchEntityDescription( key=ATTR_LED, name="Tracker LED", icon="mdi:led-on", method="async_set_led", - entity_category=ENTITY_CATEGORY_CONFIG, + entity_category=EntityCategory.CONFIG, ), TractiveSwitchEntityDescription( key=ATTR_LIVE_TRACKING, name="Live Tracking", icon="mdi:map-marker-path", method="async_set_live_tracking", - entity_category=ENTITY_CATEGORY_CONFIG, + entity_category=EntityCategory.CONFIG, ), )