Fix state error if offline

sbidy · Feb 10, 2021 · de5efd4 · de5efd4
1 parent da644f4
commit de5efd4
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/custom_components/wiz_light/light.py b/custom_components/wiz_light/light.py
@@ -266,12 +266,12 @@ def device_info(self):
             "model": self._bulbtype.name,
         }
 
-    async def update_state_available(self):
+    def update_state_available(self):
         """Update the state if bulb is available."""
         self._state = self._light.status
         self._available = True
 
-    async def update_state_unavailable(self):
+    def update_state_unavailable(self):
         """Update the state if bulb is unavailable."""
         self._state = False
         self._available = False
@@ -284,15 +284,15 @@ async def update_state(self):
                 _LOGGER.debug(
                     "[wizlight %s] state unavailable: %s", self._light.ip, self._state
                 )
-                await self.update_state_unavailable()
+                self.update_state_unavailable()
             else:
-                await self.update_state_available()
+                self.update_state_available()
         except TimeoutError as ex:
             _LOGGER.debug(ex)
-            await self.update_state_unavailable()
+            self.update_state_unavailable()
         except WizLightTimeOutError as ex:
             _LOGGER.debug(ex)
-            await self.update_state_unavailable()
+            self.update_state_unavailable()
         _LOGGER.debug("[wizlight %s] updated state: %s", self._light.ip, self._state)
 
     def update_brightness(self):