diff --git a/homeassistant/components/abode/switch.py b/homeassistant/components/abode/switch.py index c092c1ef3f0b48..b12615e054302e 100644 --- a/homeassistant/components/abode/switch.py +++ b/homeassistant/components/abode/switch.py @@ -11,6 +11,8 @@ _LOGGER = logging.getLogger(__name__) +DEVICE_TYPES = [CONST.TYPE_SWITCH, CONST.TYPE_VALVE] + async def async_setup_platform(hass, config, async_add_entities, discovery_info=None): """Platform uses config entry setup.""" @@ -23,8 +25,9 @@ async def async_setup_entry(hass, config_entry, async_add_entities): entities = [] - for device in data.abode.get_devices(generic_type=CONST.TYPE_SWITCH): - entities.append(AbodeSwitch(data, device)) + for device_type in DEVICE_TYPES: + for device in data.abode.get_devices(generic_type=device_type): + entities.append(AbodeSwitch(data, device)) for automation in data.abode.get_automations(generic_type=CONST.TYPE_AUTOMATION): entities.append(