From d1bb42c769f19ad543a1181977ac890377dcb394 Mon Sep 17 00:00:00 2001 From: bittles Date: Sat, 27 Apr 2024 17:32:39 -0500 Subject: [PATCH] change VacuumEntity to StateVacuumEntity --- custom_components/ecovacs/vacuum.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/custom_components/ecovacs/vacuum.py b/custom_components/ecovacs/vacuum.py index 2727f14..b5376dd 100644 --- a/custom_components/ecovacs/vacuum.py +++ b/custom_components/ecovacs/vacuum.py @@ -7,7 +7,7 @@ from typing import Any #sucks from . import sucks -from homeassistant.components.vacuum import VacuumEntity, VacuumEntityFeature +from homeassistant.components.vacuum import StateVacuumEntity, VacuumEntityFeature from homeassistant.core import HomeAssistant from homeassistant.helpers.entity_platform import AddEntitiesCallback from homeassistant.helpers.icon import icon_for_battery_level @@ -37,7 +37,7 @@ async def async_setup_platform( async_add_entities(vacuums) -class EcovacsVacuum(VacuumEntity): +class EcovacsVacuum(StateVacuumEntity): """Ecovacs Vacuums such as Deebot.""" _attr_fan_speed_list = [sucks.FAN_SPEED_NORMAL, sucks.FAN_SPEED_HIGH]