@@ -238,16 +238,16 @@ async def async_step_add_remove_device(self, user_input=None):
238
238
)
239
239
if self ._sel_device :
240
240
# Remove device from device registry
241
- device_registry = self . hass . helpers . device_registry .async_get (self .hass )
241
+ devreg = device_registry .async_get (self .hass )
242
242
243
243
conf_key = dict_get_key_or (self ._sel_device )
244
244
key = dict_get_or (self ._sel_device ).upper ()
245
- device = device_registry .async_get_device ({(DOMAIN , key )}, set ())
245
+ device = devreg .async_get_device ({(DOMAIN , key )}, set ())
246
246
if device is None :
247
247
errors [conf_key ] = "cannot_delete_device"
248
248
else :
249
249
_LOGGER .error ("Removing BLE monitor device %s from device registry" , key )
250
- device_registry .async_remove_device (device .id )
250
+ devreg .async_remove_device (device .id )
251
251
_LOGGER .error (f"Removing BLE monitor device %s from configuration { device } " , key )
252
252
del self ._devices [key ]
253
253
return self ._show_main_form (errors )
@@ -487,7 +487,7 @@ async def async_step_init(self, user_input=None):
487
487
)
488
488
for dev in self .config_entry .options .get (CONF_DEVICES ):
489
489
self ._devices [dict_get_or (dev ).upper ()] = dev
490
- devreg = self . hass . helpers . device_registry .async_get (self .hass )
490
+ devreg = device_registry .async_get (self .hass )
491
491
for dev in device_registry .async_entries_for_config_entry (
492
492
devreg , self .config_entry .entry_id
493
493
):
0 commit comments