From 44a31b1a03b4ca23a65d600a2f002f47e80cee73 Mon Sep 17 00:00:00 2001 From: epenet <6771947+epenet@users.noreply.github.com> Date: Wed, 6 Apr 2022 11:59:51 +0000 Subject: [PATCH] Use EntityFeature in myq --- homeassistant/components/myq/cover.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/homeassistant/components/myq/cover.py b/homeassistant/components/myq/cover.py index f3bc27a84537..861ae3790075 100644 --- a/homeassistant/components/myq/cover.py +++ b/homeassistant/components/myq/cover.py @@ -3,10 +3,9 @@ from pymyq.const import DEVICE_TYPE_GATE as MYQ_DEVICE_TYPE_GATE from pymyq.errors import MyQError from homeassistant.components.cover import ( - SUPPORT_CLOSE, - SUPPORT_OPEN, CoverDeviceClass, CoverEntity, + CoverEntityFeature, ) from homeassistant.config_entries import ConfigEntry from homeassistant.const import STATE_CLOSED, STATE_CLOSING, STATE_OPEN, STATE_OPENING @@ -36,7 +35,7 @@ async def async_setup_entry( class MyQCover(MyQEntity, CoverEntity): """Representation of a MyQ cover.""" - _attr_supported_features = SUPPORT_OPEN | SUPPORT_CLOSE + _attr_supported_features = CoverEntityFeature.OPEN | CoverEntityFeature.CLOSE def __init__(self, coordinator, device): """Initialize with API object, device id."""