From 5af31c5625a045b572f5d0d989b60c19f8fa8c68 Mon Sep 17 00:00:00 2001 From: Richard Mitchell Date: Thu, 18 Apr 2019 12:55:23 +0100 Subject: [PATCH] Fix setup --- eufy_robovac/platform.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/eufy_robovac/platform.py b/eufy_robovac/platform.py index f14907b..a41a9ef 100644 --- a/eufy_robovac/platform.py +++ b/eufy_robovac/platform.py @@ -33,13 +33,13 @@ CONFIG_SCHEMA = vol.Schema({ def setup(hass, config): """Set up Eufy devices.""" - for device_info in config[DOMAIN][CONF_DEVICES]: + for device_info in config.get(DOMAIN, {}).get(CONF_DEVICES, []): device = {} device['address'] = device_info[CONF_ADDRESS] device['local_key'] = device_info[CONF_ACCESS_TOKEN] device['device_id'] = device_info[CONF_ID] device['name'] = device_info[CONF_NAME] device['model'] = device_info[CONF_TYPE] - discovery.load_platform(hass, 'vacuum', DOMAIN, device) + discovery.load_platform(hass, 'vacuum', DOMAIN, device, config) return True