From 574d3137b9e414e6995bd503fbfcd5bd97a10d16 Mon Sep 17 00:00:00 2001
From: Tomaae <23486452+tomaae@users.noreply.github.com>
Date: Sat, 25 Jun 2022 13:56:39 +0200
Subject: [PATCH] updated devices removal entries for future use

---
 custom_components/mikrotik_router/__init__.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/custom_components/mikrotik_router/__init__.py b/custom_components/mikrotik_router/__init__.py
index 286949a..fcd8345 100644
--- a/custom_components/mikrotik_router/__init__.py
+++ b/custom_components/mikrotik_router/__init__.py
@@ -3,6 +3,8 @@
 import voluptuous as vol
 
 import homeassistant.helpers.config_validation as cv
+from homeassistant.helpers import device_registry
+from homeassistant.config_entries import ConfigEntry
 from homeassistant.exceptions import ConfigEntryNotReady
 
 from .const import (
@@ -82,6 +84,8 @@ async def async_unload_entry(hass, config_entry) -> bool:
 # ---------------------------
 #   async_remove_config_entry_device
 # ---------------------------
-async def async_remove_config_entry_device(hass, config_entry, device_entry) -> bool:
+async def async_remove_config_entry_device(
+    hass, config_entry: ConfigEntry, device_entry: device_registry.DeviceEntry
+) -> bool:
     """Remove a config entry from a device."""
     return True