Change from deprecated async_forward_entry_setup to async_forward_ent…

…ry_setups (#83)
toreamun · Dec 25, 2024 · b63b62b · b63b62b
1 parent 9ff146e
commit b63b62b
Showing 1 changed file with 3 additions and 5 deletions.
diff --git a/custom_components/amshan/__init__.py b/custom_components/amshan/__init__.py
@@ -10,7 +10,7 @@
 
 from han import common as han_type, meter_connection, obis_map
 from homeassistant import const as ha_const
-from homeassistant.const import UnitOfReactivePower
+from homeassistant.const import Platform, UnitOfReactivePower
 from homeassistant.components import sensor as ha_sensor
 from homeassistant.config_entries import ConfigEntry
 from homeassistant.core import CALLBACK_TYPE, callback, HomeAssistant, Event
@@ -27,8 +27,6 @@
 
 _LOGGER: logging.Logger = logging.getLogger(__name__)
 
-PLATFORM_TYPE = ha_const.Platform.SENSOR
-
 type AmsHanConfigEntry = ConfigEntry[AmsHanData]
 
 
@@ -140,7 +138,7 @@ async def on_hass_stop(event: Event) -> None:
 
     config_entry.runtime_data = AmsHanData(integration)
 
-    await hass.config_entries.async_forward_entry_setup(config_entry, PLATFORM_TYPE)
+    await hass.config_entries.async_forward_entry_setups(config_entry, [Platform.SENSOR])
 
     _LOGGER.debug("async_setup_entry complete.")
 
@@ -194,7 +192,7 @@ async def async_unload_entry(
 ) -> bool:
     """Handle removal of an entry."""
     is_plaform_unload_success = await hass.config_entries.async_forward_entry_unload(
-        config_entry, PLATFORM_TYPE
+        config_entry, Platform.SENSOR
     )
 
     if is_plaform_unload_success: