fix issue with changing SwitchDevice to SwitchEntity

konnected-io · Aug 2, 2020 · 92f5d50 · 92f5d50
1 parent 4d748ac
commit 92f5d50
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/custom_components/noonlight/switch.py b/custom_components/noonlight/switch.py
@@ -6,7 +6,10 @@
 from noonlight import NoonlightClient
 
 from homeassistant.components import persistent_notification
-from homeassistant.components.switch import SwitchDevice
+try:
+    from homeassistant.components.switch import SwitchDevice
+except ImportError:
+    from homeassistant.components.switch import SwitchEntity as SwitchDevice
 from homeassistant.helpers.event import async_track_time_interval
 
 from . import (DOMAIN, EVENT_NOONLIGHT_TOKEN_REFRESHED,