Fix litterrobot vacuum base class (#58765)

This commit is contained in:
Marc Mueller 2021-10-31 00:27:12 +02:00 committed by GitHub
parent f87f72bb8e
commit d6e49bc5bc
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -17,7 +17,7 @@ from homeassistant.components.vacuum import (
SUPPORT_STATUS,
SUPPORT_TURN_OFF,
SUPPORT_TURN_ON,
VacuumEntity,
StateVacuumEntity,
)
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import STATE_OFF
@ -75,7 +75,7 @@ async def async_setup_entry(
)
class LitterRobotCleaner(LitterRobotControlEntity, VacuumEntity):
class LitterRobotCleaner(LitterRobotControlEntity, StateVacuumEntity):
"""Litter-Robot "Vacuum" Cleaner."""
@property
@ -148,4 +148,5 @@ class LitterRobotCleaner(LitterRobotControlEntity, VacuumEntity):
"power_status": self.robot.power_status,
"status_code": self.robot.status_code,
"last_seen": self.robot.last_seen,
"status": self.status,
}