Merge pull request #169 from hmmbob/patch-2

Changes in setup entity platforms with group integration
Olen · Jun 6, 2024 · 9b43436 · 9b43436
2 parents 0b674dc + a1c5cdf
commit 9b43436
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/custom_components/plant/group.py b/custom_components/plant/group.py
@@ -6,10 +6,17 @@
 from homeassistant.const import STATE_OK, STATE_PROBLEM
 from homeassistant.core import HomeAssistant, callback
 
+from .const import DOMAIN
+
 
 @callback
 def async_describe_on_off_states(
     hass: HomeAssistant, registry: GroupIntegrationRegistry
 ) -> None:
     """Describe group on off states."""
-    registry.on_off_states({STATE_OK}, STATE_PROBLEM)
+    registry.on_off_states(
+        DOMAIN,
+        {STATE_OK},
+        STATE_OK,
+        STATE_PROBLEM,
+    )