fix: fix for missing entities

Author: MindFreeze

src/chart.ts

@@ -415,6 +415,9 @@ export class Chart extends LitElement {
     }
 
     let entity = this.states[getEntityId(entityConf)];
+    if (!entity) {
+      throw new Error('Entity not found ' + getEntityId(entityConf));
+    }
 
     if (entityConf.type === 'passthrough') {
       const connections = this.connectionsByChild.get(entityConf);

src/ha-sankey-chart.ts

@@ -73,7 +73,9 @@ export class SankeyChart extends SubscribeMixin(LitElement) {
           const stats = await getStatistics(this.hass, data, this.entityIds);
           const states: HassEntities = {};
           Object.keys(stats).forEach(id => {
-            states[id] = {...this.hass.states[id], state: String(stats[id])};
+            if (this.hass.states[id]) {
+              states[id] = {...this.hass.states[id], state: String(stats[id])};
+            }
           });
           this.states = states;
           // this.requestUpdate();