Deduplicate entity types

dan-r · Jan 15, 2024 · 5977532 · 5977532
1 parent 713d7ec
commit 5977532
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/custom_components/ohme/__init__.py b/custom_components/ohme/__init__.py
@@ -65,8 +65,7 @@ async def async_setup_entry(hass, entry):
     hass.data[DOMAIN][DATA_COORDINATORS] = coordinators
 
     # Create tasks for each entity type
-    entity_types = ["sensor", "binary_sensor", "switch", "button", "number", "time"]
-    for entity_type in entity_types:
+    for entity_type in ENTITY_TYPES:
         hass.async_create_task(
             hass.config_entries.async_forward_entry_setup(entry, entity_type)
         )
@@ -77,4 +76,4 @@ async def async_setup_entry(hass, entry):
 async def async_unload_entry(hass, entry):
     """Unload a config entry."""
 
-    return await hass.config_entries.async_unload_platforms(entry, ['binary_sensor', 'sensor', 'switch'])
+    return await hass.config_entries.async_unload_platforms(entry, ENTITY_TYPES)