From 9332b255f88be40e7b2b39c207f301f3d6d795f0 Mon Sep 17 00:00:00 2001
From: Snuffy2 <Snuffy2@users.noreply.github.com>
Date: Sun, 27 Oct 2024 19:39:05 -0400
Subject: [PATCH] Fix Suggested Unit of Measurement ValueError

---
 custom_components/variable/sensor.py | 16 +++++++++++-----
 1 file changed, 11 insertions(+), 5 deletions(-)

diff --git a/custom_components/variable/sensor.py b/custom_components/variable/sensor.py
index 1263213..5ab5b80 100644
--- a/custom_components/variable/sensor.py
+++ b/custom_components/variable/sensor.py
@@ -1,10 +1,13 @@
-from collections.abc import MutableMapping
 import copy
 import logging
+from collections.abc import MutableMapping
 
+import homeassistant.helpers.entity_registry as er
+import voluptuous as vol
 from homeassistant.components.sensor import (
     CONF_STATE_CLASS,
     PLATFORM_SCHEMA,
+    UNIT_CONVERTERS,
     RestoreSensor,
 )
 from homeassistant.config_entries import ConfigEntry
@@ -19,11 +22,10 @@
     Platform,
 )
 from homeassistant.core import HomeAssistant
-from homeassistant.helpers import config_validation as cv, entity_platform
+from homeassistant.helpers import config_validation as cv
+from homeassistant.helpers import entity_platform
 from homeassistant.helpers.entity import generate_entity_id
-import homeassistant.helpers.entity_registry as er
 from homeassistant.util import slugify
-import voluptuous as vol
 
 from .const import (
     ATTR_ATTRIBUTES,
@@ -156,7 +158,7 @@ def __init__(
         self._value_type = config.get(CONF_VALUE_TYPE)
         self._attr_device_class = config.get(CONF_DEVICE_CLASS)
         self._attr_native_unit_of_measurement = config.get(CONF_UNIT_OF_MEASUREMENT)
-        self._attr_suggested_unit_of_measurement = config.get(CONF_UNIT_OF_MEASUREMENT)
+        self._attr_suggested_unit_of_measurement = None
         self._attr_state_class = config.get(CONF_STATE_CLASS)
         if (
             config.get(CONF_ATTRIBUTES) is not None
@@ -180,6 +182,10 @@ def __init__(
                 )
             except ValueError:
                 self._attr_native_value = None
+        if config.get(CONF_DEVICE_CLASS) in UNIT_CONVERTERS:
+            self._attr_suggested_unit_of_measurement = config.get(
+                CONF_UNIT_OF_MEASUREMENT
+            )
 
         # _LOGGER.debug(f"({self._attr_name}) [init] unrecorded_attributes: {self._unrecorded_attributes}")