diff --git a/components/hauslane/fan.py b/components/hauslane/fan.py index 0dbe36c..54e06dc 100644 --- a/components/hauslane/fan.py +++ b/components/hauslane/fan.py @@ -1,7 +1,6 @@ import esphome.codegen as cg import esphome.config_validation as cv from esphome.components import fan, output -from esphome.const import CONF_OUTPUT_ID from . import ( HAUSLANE_SCHEMA, CONF_HAUSLANE_ID, @@ -16,15 +15,14 @@ HauslaneFan = hauslane_ns.class_( ) CONFIG_SCHEMA = ( - fan.fan_schema(Fan).extend({ + fan.fan_schema(SpeedFan).extend({ cv.GenerateID(CONF_OUTPUT_ID): cv.declare_id(HauslaneFan), }) .extend(HAUSLANE_SCHEMA) ) async def to_code(config): - var = cg.new_Pvariable(config[CONF_OUTPUT_ID]) - await fan.register_fan(var,config) + var = await fan.new_fan(config) paren = await cg.get_variable(config[CONF_HAUSLANE_ID]) cg.add(var.set_parent(paren))