Merge pull request #7 from elad-bar/main

Fix issues #4 and #5
rroller · Aug 23, 2021 · 25fe6a5 · 25fe6a5
2 parents 5df5e79 + 1c27a39
commit 25fe6a5
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/custom_components/netgear_wax/__init__.py b/custom_components/netgear_wax/__init__.py
@@ -163,7 +163,7 @@ def get_stats(self) -> Dict[str, Stat]:
 async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
     """Handle removal of an entry."""
     coordinator = hass.data[DOMAIN][entry.entry_id]
-    await coordinator.async_stop()
+    await coordinator.async_stop({})
     unloaded = all(
         await asyncio.gather(
             *[

diff --git a/custom_components/netgear_wax/binary_sensor.py b/custom_components/netgear_wax/binary_sensor.py
@@ -5,6 +5,7 @@
 from homeassistant.components.binary_sensor import BinarySensorEntity
 from homeassistant.core import HomeAssistant
 
+from . import NetgearDataUpdateCoordinator
 from .const import (
     DOMAIN, SAFETY_DEVICE_CLASS,
 )