From 172591031dce626b3678ccb227c0731c166bbca8 Mon Sep 17 00:00:00 2001 From: epenet <6771947+epenet@users.noreply.github.com> Date: Thu, 9 Dec 2021 09:18:20 +0100 Subject: [PATCH] Use new CoverDeviceClass enum in brunt (#61328) Co-authored-by: epenet --- homeassistant/components/brunt/cover.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/homeassistant/components/brunt/cover.py b/homeassistant/components/brunt/cover.py index cc0ecd0feab..faa4653d3be 100644 --- a/homeassistant/components/brunt/cover.py +++ b/homeassistant/components/brunt/cover.py @@ -10,10 +10,10 @@ from brunt import BruntClientAsync, Thing from homeassistant.components.cover import ( ATTR_POSITION, - DEVICE_CLASS_SHADE, SUPPORT_CLOSE, SUPPORT_OPEN, SUPPORT_SET_POSITION, + CoverDeviceClass, CoverEntity, ) from homeassistant.config_entries import SOURCE_IMPORT, ConfigEntry @@ -101,7 +101,7 @@ class BruntDevice(CoordinatorEntity, CoverEntity): self._remove_update_listener = None self._attr_name = self._thing.NAME - self._attr_device_class = DEVICE_CLASS_SHADE + self._attr_device_class = CoverDeviceClass.SHADE self._attr_supported_features = COVER_FEATURES self._attr_attribution = ATTRIBUTION self._attr_device_info = DeviceInfo(