From ede8d1157072bc08571b9b76d3cf3ff66c56da05 Mon Sep 17 00:00:00 2001
From: MapoDan <42698485+MapoDan@users.noreply.github.com>
Date: Tue, 2 Jun 2020 16:34:36 +0200
Subject: [PATCH] Update climate.py

---
 custom_components/programmable_thermostat/climate.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/custom_components/programmable_thermostat/climate.py b/custom_components/programmable_thermostat/climate.py
index 3325176..71b0e33 100644
--- 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):