From 7b5c80720a9415f89d1076e62adfbdd0b6c543e0 Mon Sep 17 00:00:00 2001 From: Erik Montnemery Date: Fri, 25 Nov 2022 20:12:27 +0100 Subject: [PATCH] Tweak scaffold script (#82705) --- .../templates/config_flow_helper/integration/config_flow.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/script/scaffold/templates/config_flow_helper/integration/config_flow.py b/script/scaffold/templates/config_flow_helper/integration/config_flow.py index 015aac2ca91..f9b8e81d364 100644 --- a/script/scaffold/templates/config_flow_helper/integration/config_flow.py +++ b/script/scaffold/templates/config_flow_helper/integration/config_flow.py @@ -6,6 +6,7 @@ from typing import Any, cast import voluptuous as vol +from homeassistant.components.sensor import DOMAIN as SENSOR_DOMAIN from homeassistant.const import CONF_ENTITY_ID from homeassistant.helpers import selector from homeassistant.helpers.schema_config_entry_flow import ( @@ -19,7 +20,7 @@ from .const import DOMAIN OPTIONS_SCHEMA = vol.Schema( { vol.Required(CONF_ENTITY_ID): selector.EntitySelector( - selector.EntitySelectorConfig(domain="sensor") + selector.EntitySelectorConfig(domain=SENSOR_DOMAIN) ), } )