diff --git a/esphome/components/udp/sensor.py b/esphome/components/udp/sensor.py index 931ce0a116..4519bc7709 100644 --- a/esphome/components/udp/sensor.py +++ b/esphome/components/udp/sensor.py @@ -1,5 +1,7 @@ import esphome.codegen as cg -from esphome.components import sensor +from esphome.const import CONF_ID +from esphome.config_validation import has_at_least_one_key, All +from esphome.components.sensor import sensor_schema, new_sensor from . import ( SENSOR_SCHEMA, CONF_UDP_ID, @@ -7,20 +9,18 @@ from . import ( CONF_PROVIDER, require_internal_with_name, ) -from ...config_validation import has_at_least_one_key, All -from ...const import CONF_ID AUTO_LOAD = ["udp"] CONFIG_SCHEMA = All( - sensor.sensor_schema().extend(SENSOR_SCHEMA), + sensor_schema().extend(SENSOR_SCHEMA), has_at_least_one_key(CONF_ID, CONF_REMOTE_ID), require_internal_with_name, ) async def to_code(config): - var = await sensor.new_sensor(config) + var = await new_sensor(config) comp = await cg.get_variable(config[CONF_UDP_ID]) remote_id = str(config.get(CONF_REMOTE_ID) or config.get(CONF_ID)) cg.add(comp.add_remote_sensor(config[CONF_PROVIDER], remote_id, var))