diff --git a/miio/integrations/roborock/vacuum/vacuumcontainers.py b/miio/integrations/roborock/vacuum/vacuumcontainers.py index 65a5963a3..ea42451f2 100644 --- a/miio/integrations/roborock/vacuum/vacuumcontainers.py +++ b/miio/integrations/roborock/vacuum/vacuumcontainers.py @@ -228,6 +228,8 @@ def error_code(self) -> int: "Error string", id=VacuumId.ErrorMessage, icon="mdi:alert", + device_class="enum", + options=list(ERROR_CODES.values()), entity_category="diagnostic", enabled_default=False, ) @@ -255,6 +257,8 @@ def dock_error_code(self) -> Optional[int]: @sensor( "Dock error string", icon="mdi:alert", + device_class="enum", + options=list(dock_error_codes.values()), entity_category="diagnostic", enabled_default=False, )