Fix for Keyerror 'mode' issue

gogades · Mar 26, 2020 · f78bf0a · f78bf0a
1 parent 9244a99
commit f78bf0a
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/custom_components/infinitive/climate.py b/custom_components/infinitive/climate.py
@@ -254,7 +254,9 @@ def hvac_action(self):
     def update(self):
         """Update current status from infinitive device."""
         _LOGGER.debug("Updating Infinitive status")
-        self._status = self._inf_device.get_status()
+        status = self._inf_device.get_status()
+        if 'mode' in status.keys():
+            self._status = status
         try:
             self._hvac_mode = self._status['mode']
             self._target_temperature_high = self._status['coolSetpoint']