add PLATFORM_SCHEMAS

rlippmann · Oct 11, 2023 · e94f051 · e94f051
1 parent 9ab3043
commit e94f051
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/custom_components/adtpulse/alarm_control_panel.py b/custom_components/adtpulse/alarm_control_panel.py
@@ -20,6 +20,7 @@
 )
 from homeassistant.core import HomeAssistant, callback
 from homeassistant.exceptions import HomeAssistantError
+from homeassistant.helpers.config_validation import config_entry_only_config_schema
 from homeassistant.helpers.entity import DeviceInfo
 from homeassistant.helpers.entity_platform import (
     AddEntitiesCallback,
@@ -48,6 +49,7 @@
 
 LOG = getLogger(__name__)
 
+PLATFORM_SCHEMA = config_entry_only_config_schema
 ALARM_MAP = {
     ADT_ALARM_ARMING: STATE_ALARM_ARMING,
     ADT_ALARM_AWAY: STATE_ALARM_ARMED_AWAY,

diff --git a/custom_components/adtpulse/binary_sensor.py b/custom_components/adtpulse/binary_sensor.py
@@ -17,6 +17,7 @@
 )
 from homeassistant.config_entries import ConfigEntry
 from homeassistant.core import HomeAssistant, callback
+from homeassistant.helpers.config_validation import config_entry_only_config_schema
 from homeassistant.helpers.device_registry import CONNECTION_NETWORK_MAC
 from homeassistant.helpers.entity import DeviceInfo
 from homeassistant.helpers.entity_platform import AddEntitiesCallback
@@ -37,6 +38,8 @@
 
 LOG = getLogger(__name__)
 
+PLATFORM_SCHEMA = config_entry_only_config_schema
+
 # please keep these alphabetized to make changes easier
 ADT_DEVICE_CLASS_TAG_MAP = {
     "co": BinarySensorDeviceClass.CO,