Fix CoffeeMaker after the NumberEntity refactoring

custom_components/ge_home/devices/coffee_maker.py

@@ -60,7 +60,7 @@ class CcmApi(ApplianceApi):
     async def start_brewing(self) -> None:
         """Aggregate brew settings and start brewing."""
 
-        new_mode = ErdCcmBrewSettings(self._brew_cups_entity.value,
+        new_mode = ErdCcmBrewSettings(self._brew_cups_entity.native_value,
                                       self._brew_strengh_entity.brew_strength,
-                                      self._brew_temperature_entity.brew_temperature)
+                                      self._brew_temperature_entity.native_value)
         await self.appliance.async_set_erd_value(ErdCode.CCM_BREW_SETTINGS, new_mode)

custom_components/ge_home/entities/ccm/ge_ccm_brew_temperature.py

@@ -17,11 +17,3 @@ class GeCcmBrewTemperatureNumber(GeErdNumber, GeCcmCachedValue):
     @property
     def native_value(self):
         return int(self.get_value(device_value = super().native_value))
-    
-    @property
-    def native_unit_of_measurement(self):
-        return TEMP_FAHRENHEIT
-
-    @property
-    def brew_temperature(self) -> int:
-        return self.value

custom_components/ge_home/entities/common/ge_erd_number.py

@@ -1,12 +1,11 @@
 import logging
 from typing import Optional
 from gehomesdk.erd.erd_data_type import ErdDataType
-from homeassistant.components.number import NumberEntity
-
-from homeassistant.const import (
-    DEVICE_CLASS_TEMPERATURE,
-    TEMP_FAHRENHEIT,
+from homeassistant.components.number import (
+    NumberEntity,
+    NumberDeviceClass,
 )
+from homeassistant.const import TEMP_FAHRENHEIT
 from gehomesdk import ErdCodeType, ErdCodeClass
 from .ge_erd_entity import GeErdEntity
 from ...devices import ApplianceApi
@@ -88,7 +87,7 @@ class GeErdNumber(GeErdEntity, NumberEntity):
         if self._uom_override:
             return self._uom_override
 
-        if self.device_class == DEVICE_CLASS_TEMPERATURE:
+        if self.device_class == NumberDeviceClass.TEMPERATURE:
             #NOTE: it appears that the API only sets temperature in Fahrenheit,
             #so we'll hard code this UOM instead of using the device configured
             #settings
@@ -104,7 +103,7 @@ class GeErdNumber(GeErdEntity, NumberEntity):
             ErdCodeClass.RAW_TEMPERATURE,
             ErdCodeClass.NON_ZERO_TEMPERATURE,
         ]:
-            return DEVICE_CLASS_TEMPERATURE
+            return NumberDeviceClass.TEMPERATURE
 
         return None