diff --git a/SunGather/exports/mqtt.py b/SunGather/exports/mqtt.py index 6896d1c..93dcd43 100644 --- a/SunGather/exports/mqtt.py +++ b/SunGather/exports/mqtt.py @@ -91,7 +91,8 @@ class export_mqtt(object): config_msg['unique_id'] = "inverter_" + self.cleanName(ha_sensor.get('name')) config_msg['state_topic'] = self.mqtt_config['topic'] config_msg['value_template'] = "{{ value_json." + ha_sensor.get('register') + " }}" - config_msg['unit_of_measurement'] = inverter.getRegisterUnit(ha_sensor.get('register')) + if inverter.getRegisterUnit(ha_sensor.get('register')): + config_msg['unit_of_measurement'] = inverter.getRegisterUnit(ha_sensor.get('register')) if ha_sensor.get('dev_class'): config_msg['device_class'] = ha_sensor.get('dev_class') if ha_sensor.get('state_class'): @@ -108,4 +109,4 @@ class export_mqtt(object): self.ha_discovery_published = True logging.info("MQTT: Published Home Assistant Discovery messages") - return True \ No newline at end of file + return True