From 63692bba677a19f497cdb4ff1cfcb0d44a0f6a35 Mon Sep 17 00:00:00 2001
From: Paulus Schoutsen <paulus.schoutsen@mycase.com>
Date: Sun, 23 Aug 2015 10:17:45 -0700
Subject: [PATCH] Tweak sensor label babdge

---
 src/components/entity/ha-state-label-badge.js | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/src/components/entity/ha-state-label-badge.js b/src/components/entity/ha-state-label-badge.js
index 1df4d3d206..bb5989f1e0 100644
--- a/src/components/entity/ha-state-label-badge.js
+++ b/src/components/entity/ha-state-label-badge.js
@@ -67,6 +67,8 @@ export default new Polymer({
     case 'scene':
     case 'script':
       return undefined;
+    case 'sensor':
+      return state.attributes.unit_of_measurement && state.state;
     default:
       return state.state;
     }
@@ -77,6 +79,8 @@ export default new Polymer({
     case 'scene':
     case 'script':
       return domainIcon(state.domain);
+    case 'sensor':
+      return !state.attributes.unit_of_measurement && domainIcon(state.domain);
     case 'sun':
       return state.state === 'above_horizon' ?
         'image:wb-sunny' : 'image:brightness-3';
@@ -99,6 +103,8 @@ export default new Polymer({
     case 'scene':
     case 'script':
       return state.domain;
+    case 'sensor':
+      return !state.attributes.unit_of_measurement && state.state;
     case 'device_tracker':
       return state.state === 'home' ? 'Home' : 'Away';
     default: