From 360c625e1501536d7db0f4253aed15a92a852369 Mon Sep 17 00:00:00 2001 From: magico13 Date: Sun, 5 Nov 2023 12:42:08 -0500 Subject: [PATCH] Add awaits to service call (untested) --- custom_components/emporia_vue/__init__.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/custom_components/emporia_vue/__init__.py b/custom_components/emporia_vue/__init__.py index 2b8400d..a567f33 100644 --- a/custom_components/emporia_vue/__init__.py +++ b/custom_components/emporia_vue/__init__.py @@ -225,8 +225,8 @@ async def handle_set_charger_current(call): charger_entity = None if device_id: - entity_registry = er.async_get(hass) - entities = er.async_entries_for_device(entity_registry, device_id[0]) + entity_registry = await er.async_get(hass) + entities = await er.async_entries_for_device(entity_registry, device_id[0]) for entity in entities: _LOGGER.info("Entity is %s", str(entity)) if entity.entity_id.startswith("switch"): @@ -235,8 +235,8 @@ async def handle_set_charger_current(call): if not charger_entity: charger_entity = entities[0] elif entity_id: - entity_registry = er.async_get(hass) - charger_entity = entity_registry.async_get(entity_id[0]) + entity_registry = await er.async_get(hass) + charger_entity = await entity_registry.async_get(entity_id[0]) if not charger_entity: raise HomeAssistantError("Target device or Entity required.")