diff --git a/homeassistant/components/xiaomi_miio/const.py b/homeassistant/components/xiaomi_miio/const.py index 6630def38ef..578a6d3ffab 100644 --- a/homeassistant/components/xiaomi_miio/const.py +++ b/homeassistant/components/xiaomi_miio/const.py @@ -199,8 +199,10 @@ MODELS_LIGHT = ( # TODO: use const from pythonmiio once new release with the constant has been published. # pylint: disable=fixme ROCKROBO_S4 = "roborock.vacuum.s4" ROCKROBO_S5_MAX = "roborock.vacuum.s5e" +ROCKROBO_E2 = "roborock.vacuum.e2" MODELS_VACUUM = [ ROCKROBO_V1, + ROCKROBO_E2, ROCKROBO_S4, ROCKROBO_S5, ROCKROBO_S5_MAX, @@ -209,6 +211,7 @@ MODELS_VACUUM = [ ROCKROBO_S7, ] MODELS_VACUUM_WITH_MOP = [ + ROCKROBO_E2, ROCKROBO_S5, ROCKROBO_S5_MAX, ROCKROBO_S6,