Update climate.py

custom-components · Jun 2, 2020 · ede8d11 · ede8d11
1 parent 3e0680f
commit ede8d11
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/custom_components/programmable_thermostat/climate.py b/custom_components/programmable_thermostat/climate.py
@@ -4,7 +4,7 @@
 import json
 
 from homeassistant.config_entries import SOURCE_IMPORT
-from homeassistant.components.climate import PLATFORM_SCHEMA, ClimateDevice
+from homeassistant.components.climate import PLATFORM_SCHEMA, ClimateEntity
 from homeassistant.components.climate.const import (
     CURRENT_HVAC_COOL,
     CURRENT_HVAC_HEAT,
@@ -75,7 +75,7 @@ async def async_setup_entry(hass, config_entry, async_add_devices):
     async_add_devices([ProgrammableThermostat(hass, config_entry.data)])
 
 
-class ProgrammableThermostat(ClimateDevice, RestoreEntity):
+class ProgrammableThermostat(ClimateEntity, RestoreEntity):
     """ProgrammableThermostat."""
 
     def __init__(self, hass, config):