From 232d79366199a6dcf6e0573ad81e418c59e2a599 Mon Sep 17 00:00:00 2001 From: epenet <6771947+epenet@users.noreply.github.com> Date: Wed, 15 Dec 2021 15:17:18 +0100 Subject: [PATCH] Use new DeviceClass enums in lutron_caseta (#61887) Co-authored-by: epenet --- homeassistant/components/lutron_caseta/binary_sensor.py | 4 ++-- homeassistant/components/lutron_caseta/cover.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/homeassistant/components/lutron_caseta/binary_sensor.py b/homeassistant/components/lutron_caseta/binary_sensor.py index c2fc311de43..e0966cd0f42 100644 --- a/homeassistant/components/lutron_caseta/binary_sensor.py +++ b/homeassistant/components/lutron_caseta/binary_sensor.py @@ -2,7 +2,7 @@ from pylutron_caseta import OCCUPANCY_GROUP_OCCUPIED from homeassistant.components.binary_sensor import ( - DEVICE_CLASS_OCCUPANCY, + BinarySensorDeviceClass, BinarySensorEntity, ) @@ -35,7 +35,7 @@ class LutronOccupancySensor(LutronCasetaDevice, BinarySensorEntity): @property def device_class(self): """Flag supported features.""" - return DEVICE_CLASS_OCCUPANCY + return BinarySensorDeviceClass.OCCUPANCY @property def is_on(self): diff --git a/homeassistant/components/lutron_caseta/cover.py b/homeassistant/components/lutron_caseta/cover.py index 31f7e9b55bd..111da12a6f5 100644 --- a/homeassistant/components/lutron_caseta/cover.py +++ b/homeassistant/components/lutron_caseta/cover.py @@ -3,12 +3,12 @@ import logging from homeassistant.components.cover import ( ATTR_POSITION, - DEVICE_CLASS_SHADE, DOMAIN, SUPPORT_CLOSE, SUPPORT_OPEN, SUPPORT_SET_POSITION, SUPPORT_STOP, + CoverDeviceClass, CoverEntity, ) @@ -58,7 +58,7 @@ class LutronCasetaCover(LutronCasetaDevice, CoverEntity): @property def device_class(self): """Return the device class.""" - return DEVICE_CLASS_SHADE + return CoverDeviceClass.SHADE async def async_stop_cover(self, **kwargs): """Top the cover."""