Only look for sensors not binary for batteries

andrew-codechimp · Jan 26, 2024 · 112aea7 · 112aea7
1 parent 9f8dda4
commit 112aea7
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/custom_components/battery_notes/device.py b/custom_components/battery_notes/device.py
@@ -85,10 +85,7 @@ async def async_setup(self) -> bool:
         for entity in entity_registry.entities.values():
             if not entity.device_id or entity.device_id != device_id:
                 continue
-            if not entity.domain or entity.domain not in {
-                BINARY_SENSOR_DOMAIN,
-                SENSOR_DOMAIN,
-            }:
+            if not entity.domain or entity.domain != SENSOR_DOMAIN:
                 continue
             if not entity.platform or entity.platform == DOMAIN:
                 continue