Update switch.py

MizterB · Oct 13, 2023 · 2e0e106 · 2e0e106
1 parent 11a2bc6
commit 2e0e106
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/custom_components/phyn/switch.py b/custom_components/phyn/switch.py
@@ -27,7 +27,8 @@ async def async_setup_entry(
     ]["devices"]
     entities = []
     for device in devices:
-        entities.append(PhynSwitch(device))
+        if device.model == 'PP1' or device.model == 'PP2':
+            entities.append(PhynSwitch(device))
     async_add_entities(entities)