Fix warning about import.

boralyl · May 23, 2020 · 278c598 · 278c598
1 parent e23818f
commit 278c598
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/custom_components/steam_wishlist/entities.py b/custom_components/steam_wishlist/entities.py
@@ -1,7 +1,11 @@
 import logging
 from typing import List
 
-from homeassistant.components.binary_sensor import BinarySensorDevice
+try:
+    from homeassistant.components.binary_sensor import BinarySensorEntity
+except ImportError:
+    # Prior to HA v0.110
+    from homeassistant.components.binary_sensor import BinarySensorDevice
 from homeassistant.helpers.entity import Entity
 from homeassistant.util import slugify