Cleanup deprecated async_get_registry

zachcheatham · Jun 9, 2022 · f2e56f5 · f2e56f5
1 parent 60b415b
commit f2e56f5
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/custom_components/omada/device_tracker.py b/custom_components/omada/device_tracker.py
@@ -63,7 +63,7 @@ def items_added(macs=None):
         async_dispatcher_connect(hass, controller.signal_update, items_added)
     )
 
-    entity_registry = await hass.helpers.entity_registry.async_get_registry()
+    entity_registry = await hass.helpers.entity_registry.async_get(hass)
     initial_set = set()
 
     # Add connected entries
@@ -173,7 +173,7 @@ def should_poll(self) -> bool:
         return False
 
     async def remove(self):
-        entity_registry = await self.hass.helpers.entity_registry.async_get_registry()
+        entity_registry = await self.hass.helpers.entity_registry.async_get(self.hass)
 
         await self.async_remove()
         entity_registry.async_remove(self.entity_id)
@@ -314,7 +314,7 @@ def site(self) -> str:
         return site
 
     async def remove(self):
-        entity_registry = await self.hass.helpers.entity_registry.async_get_registry()
+        entity_registry = await self.hass.helpers.entity_registry.async_get(self.hass)
 
         await self.async_remove()
         entity_registry.async_remove(self.entity_id)