Add class as first positional arg to sensor_schema (#3216)

esphome/components/tsl2561/sensor.py

@@ -3,7 +3,6 @@ import esphome.config_validation as cv
 from esphome.components import i2c, sensor
 from esphome.const import (
     CONF_GAIN,
-    CONF_ID,
     CONF_INTEGRATION_TIME,
     DEVICE_CLASS_ILLUMINANCE,
     STATE_CLASS_MEASUREMENT,
@@ -40,6 +39,7 @@ TSL2561Sensor = tsl2561_ns.class_(
 
 CONFIG_SCHEMA = (
     sensor.sensor_schema(
+        TSL2561Sensor,
         unit_of_measurement=UNIT_LUX,
         accuracy_decimals=1,
         device_class=DEVICE_CLASS_ILLUMINANCE,
@@ -47,7 +47,6 @@ CONFIG_SCHEMA = (
     )
     .extend(
         {
-            cv.GenerateID(): cv.declare_id(TSL2561Sensor),
             cv.Optional(
                 CONF_INTEGRATION_TIME, default="402ms"
             ): validate_integration_time,
@@ -61,10 +60,9 @@ CONFIG_SCHEMA = (
 
 
 async def to_code(config):
-    var = cg.new_Pvariable(config[CONF_ID])
+    var = await sensor.new_sensor(config)
     await cg.register_component(var, config)
     await i2c.register_i2c_device(var, config)
-    await sensor.register_sensor(var, config)
 
     cg.add(var.set_integration_time(config[CONF_INTEGRATION_TIME]))
     cg.add(var.set_gain(config[CONF_GAIN]))