config_entry forwarded only to proper platform; DOMAIN imported from component

custom_components/localtuya/switch.py

@@ -32,6 +32,7 @@ import voluptuous as vol
 
 from homeassistant.components.switch import (
     SwitchEntity,
+    DOMAIN,
     PLATFORM_SCHEMA,
 )
 from homeassistant.const import (
@@ -55,8 +56,6 @@ from .pytuya import TuyaDevice
 
 _LOGGER = logging.getLogger(__name__)
 
-PLATFORM = "switch"
-
 DEFAULT_ID = "1"
 
 # TODO: This will eventully merge with flow_schema
@@ -93,7 +92,7 @@ def flow_schema(dps):
 async def async_setup_entry(hass, config_entry, async_add_entities):
     """Setup a Tuya switch based on a config entry."""
     device, entities_to_setup = prepare_setup_entities(
-        config_entry, PLATFORM
+        config_entry, DOMAIN
     )
     if not entities_to_setup:
         return
@@ -116,7 +115,7 @@ async def async_setup_entry(hass, config_entry, async_add_entities):
 
 def setup_platform(hass, config, add_devices, discovery_info=None):
     """Set up of the Tuya switch."""
-    return import_from_yaml(hass, config, PLATFORM)
+    return import_from_yaml(hass, config, DOMAIN)
 
 
 class TuyaCache: