Tweak sensor state to handle missing attributes (#76)

cgarwood · Dec 28, 2021 · 1cae445 · 1cae445
1 parent bd00731
commit 1cae445
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/custom_components/fullykiosk/sensor.py b/custom_components/fullykiosk/sensor.py
@@ -110,7 +110,7 @@ def state(self):
         if self._sensor in STORAGE_SENSORS:
             return round(self.coordinator.data[self._sensor] * 0.000001, 1)
 
-        return self.coordinator.data[self._sensor]
+        return self.coordinator.data.get(self._sensor)
 
     async def async_added_to_hass(self):
         """Connect to dispatcher listening for entity data notifications."""