Merge pull request #108 from edenhaus/patch-1

Update BinarySensorDevice to BinarySensorEntity
custom-components · Nov 16, 2020 · f49338f · f49338f
2 parents 4a1618b + 8520c5f
commit f49338f
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/custom_components/grocy/binary_sensor.py b/custom_components/grocy/binary_sensor.py
@@ -1,6 +1,11 @@
 """Binary sensor platform for Grocy."""
 import logging
-from homeassistant.components.binary_sensor import BinarySensorDevice
+try:
+    from homeassistant.components.binary_sensor import BinarySensorEntity
+except ImportError:
+    from homeassistant.components.binary_sensor import (
+        BinarySensorDevice as BinarySensorEntity,
+    )
 
 # pylint: disable=relative-beyond-top-level
 from .const import (
@@ -33,7 +38,7 @@ async def async_setup_entry(hass, entry, async_add_entities):
     async_add_entities(entities, True)
 
 
-class GrocyBinarySensor(GrocyEntity, BinarySensorDevice):
+class GrocyBinarySensor(GrocyEntity, BinarySensorEntity):
     """Grocy binary_sensor class."""
 
     @property