diff --git a/homeassistant/components/modbus/base_platform.py b/homeassistant/components/modbus/base_platform.py index d9ee58f3c38..efcb70b5b16 100644 --- a/homeassistant/components/modbus/base_platform.py +++ b/homeassistant/components/modbus/base_platform.py @@ -27,7 +27,9 @@ from homeassistant.helpers.restore_state import RestoreEntity from .const import ( CALL_TYPE_COIL, + CALL_TYPE_DISCRETE, CALL_TYPE_REGISTER_HOLDING, + CALL_TYPE_REGISTER_INPUT, CALL_TYPE_WRITE_COIL, CALL_TYPE_WRITE_COILS, CALL_TYPE_WRITE_REGISTER, @@ -173,6 +175,14 @@ class BaseSwitch(BasePlatform, ToggleEntity, RestoreEntity): CALL_TYPE_REGISTER_HOLDING, CALL_TYPE_WRITE_REGISTER, ), + CALL_TYPE_DISCRETE: ( + CALL_TYPE_DISCRETE, + None, + ), + CALL_TYPE_REGISTER_INPUT: ( + CALL_TYPE_REGISTER_INPUT, + None, + ), CALL_TYPE_COIL: (CALL_TYPE_COIL, CALL_TYPE_WRITE_COIL), CALL_TYPE_X_COILS: (CALL_TYPE_COIL, CALL_TYPE_WRITE_COILS), CALL_TYPE_X_REGISTER_HOLDINGS: (