WIP

esphome/components/fyrtur_motor/switch/__init__.py

@@ -2,7 +2,7 @@ import esphome.codegen as cg
 from esphome.components import switch
 import esphome.config_validation as cv
 from esphome.const import (
-    DEVICE_CLASS_SWITCH,
+    ENTITY_CATEGORY_CONFIG,
     ICON_ARROW_EXPAND_VERTICAL,
 )
 from .. import CONF_FYRTUR_MOTOR_ID, FyrturMotorComponent, fyrtur_motor_ns
@@ -15,7 +15,7 @@ CONFIG_SCHEMA = {
     cv.GenerateID(CONF_FYRTUR_MOTOR_ID): cv.use_id(FyrturMotorComponent),
     cv.Required(CONF_OPEN_CLOSE): switch.switch_schema(
         OpenCloseSwitch,
-        device_class=DEVICE_CLASS_SWITCH,
+        entity_category=ENTITY_CATEGORY_CONFIG,
         icon=ICON_ARROW_EXPAND_VERTICAL,
     ),
 }