Add missing CONFIG_SCHEMA

wbyoung · Sep 16, 2024 · 9f26347 · 9f26347
1 parent ff51a3b
commit 9f26347
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/custom_components/watersmart/__init__.py b/custom_components/watersmart/__init__.py
@@ -5,6 +5,7 @@
 from homeassistant.const import CONF_HOST, CONF_PASSWORD, CONF_USERNAME, Platform
 from homeassistant.core import HomeAssistant
 from homeassistant.helpers.aiohttp_client import async_get_clientsession
+from homeassistant.helpers import config_validation as cv
 from homeassistant.helpers.typing import ConfigType
 
 from .client import WaterSmartClient
@@ -14,6 +15,7 @@
 from .types import WaterSmartConfigEntry, WaterSmartData
 
 PLATFORMS: list[Platform] = [Platform.SENSOR]
+CONFIG_SCHEMA = cv.config_entry_only_config_schema(DOMAIN)
 
 
 async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool: