From ca785266b453a70d75ba528f88c652754e5a1340 Mon Sep 17 00:00:00 2001 From: epenet <6771947+epenet@users.noreply.github.com> Date: Wed, 18 May 2022 13:42:49 +0200 Subject: [PATCH] Cleanup deprecated async_get_registry in hue (#72068) --- homeassistant/components/hue/v1/helpers.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/homeassistant/components/hue/v1/helpers.py b/homeassistant/components/hue/v1/helpers.py index d4582f0bb52..b0d774915df 100644 --- a/homeassistant/components/hue/v1/helpers.py +++ b/homeassistant/components/hue/v1/helpers.py @@ -1,7 +1,6 @@ """Helper functions for Philips Hue.""" -from homeassistant.helpers.device_registry import async_get_registry as get_dev_reg -from homeassistant.helpers.entity_registry import async_get_registry as get_ent_reg +from homeassistant.helpers import device_registry as dr, entity_registry as er from ..const import DOMAIN @@ -18,10 +17,10 @@ async def remove_devices(bridge, api_ids, current): entity = current[item_id] removed_items.append(item_id) await entity.async_remove(force_remove=True) - ent_registry = await get_ent_reg(bridge.hass) + ent_registry = er.async_get(bridge.hass) if entity.entity_id in ent_registry.entities: ent_registry.async_remove(entity.entity_id) - dev_registry = await get_dev_reg(bridge.hass) + dev_registry = dr.async_get(bridge.hass) device = dev_registry.async_get_device(identifiers={(DOMAIN, entity.device_id)}) if device is not None: dev_registry.async_update_device(