Fixing HomeAssistantType from @dcmeglio

skweeker · Jan 4, 2025 · 4e9e391 · 4e9e391
1 parent 1028132
commit 4e9e391
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/custom_components/calendarific/__init__.py b/custom_components/calendarific/__init__.py
@@ -7,6 +7,7 @@
 import voluptuous as vol
 
 from homeassistant.config_entries import ConfigEntry
+from homeassistant.const import Platform
 from homeassistant.core import HomeAssistant
 
 from homeassistant import config_entries
@@ -54,14 +55,12 @@ def setup(hass, config):
 
 
 async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry):
-    hass.async_create_task(
-        hass.config_entries.async_forward_entry_setup(entry, "sensor")
-    )
+    await hass.config_entries.async_forward_entry_setups(entry, [Platform.SENSOR])
     return True
 
 async def async_unload_entry(hass, entry):
     """Unload a config entry."""
-    return await hass.config_entries.async_forward_entry_unload(entry, "sensor")
+    return await hass.config_entries.async_forward_entry_unload(entry, Platform.SENSOR)
 
 class CalendarificApiReader: