use entity instead of device

piitaya · Jun 9, 2020 · c9e106f · c9e106f
1 parent 53d3005
commit c9e106f
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/custom_components/qubino_wire_pilot/climate.py b/custom_components/qubino_wire_pilot/climate.py
@@ -5,7 +5,12 @@
 
 import voluptuous as vol
 
-from homeassistant.components.climate import ClimateDevice, PLATFORM_SCHEMA
+from homeassistant.components.climate import PLATFORM_SCHEMA
+try:
+    from homeassistant.components.climate import ClimateEntity
+except ImportError:
+    from homeassistant.components.climate import ClimateDevice as ClimateEntity
+
 from homeassistant.components.climate.const import (
     SUPPORT_PRESET_MODE,
     HVAC_MODE_AUTO,
@@ -71,7 +76,7 @@ async def async_setup_platform(hass, config, async_add_entities, discovery_info=
     )
 
 
-class QubinoWirePilotClimate(ClimateDevice, RestoreEntity):
+class QubinoWirePilotClimate(ClimateEntity, RestoreEntity):
     """Representation of a Qubino Wire Pilot device."""
 
     def __init__(self, name, heater_entity_id, sensor_entity_id):