From 28b6306ad625af598aef52619bc07159ca803bd0 Mon Sep 17 00:00:00 2001 From: Franck Nijhof Date: Sat, 4 Feb 2023 00:08:15 +0100 Subject: [PATCH] Remove use of Number DEVICE_CLASSES_SCHEMA from MQTT (#87345) --- homeassistant/components/mqtt/number.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/homeassistant/components/mqtt/number.py b/homeassistant/components/mqtt/number.py index 3682b19cf4e..62bb9123a76 100644 --- a/homeassistant/components/mqtt/number.py +++ b/homeassistant/components/mqtt/number.py @@ -12,7 +12,7 @@ from homeassistant.components.number import ( DEFAULT_MAX_VALUE, DEFAULT_MIN_VALUE, DEFAULT_STEP, - DEVICE_CLASSES_SCHEMA, + NumberDeviceClass, NumberMode, RestoreNumber, ) @@ -86,7 +86,9 @@ def validate_config(config: ConfigType) -> ConfigType: _PLATFORM_SCHEMA_BASE = MQTT_RW_SCHEMA.extend( { vol.Optional(CONF_COMMAND_TEMPLATE): cv.template, - vol.Optional(CONF_DEVICE_CLASS): vol.Any(DEVICE_CLASSES_SCHEMA, None), + vol.Optional(CONF_DEVICE_CLASS): vol.Any( + vol.All(vol.Lower, vol.Coerce(NumberDeviceClass)), None + ), vol.Optional(CONF_MAX, default=DEFAULT_MAX_VALUE): vol.Coerce(float), vol.Optional(CONF_MIN, default=DEFAULT_MIN_VALUE): vol.Coerce(float), vol.Optional(CONF_MODE, default=NumberMode.AUTO): vol.Coerce(NumberMode),