diff --git a/custom_components/kobrax_lan/sensor.py b/custom_components/kobrax_lan/sensor.py index e620525..3c1bb47 100644 --- a/custom_components/kobrax_lan/sensor.py +++ b/custom_components/kobrax_lan/sensor.py @@ -523,6 +523,7 @@ async def async_setup_entry(hass, entry, async_add_entities): for ace_index in range(MAX_ACE_UNITS): # Switch entities ace_entities.append(("switch", f"{entry.entry_id}_ace_{ace_index}_auto_feed")) + ace_entities.append(("switch", f"{entry.entry_id}_ace_{ace_index}_dryer")) # Number entities (temp + duration) ace_entities.append(("number", f"{entry.entry_id}_ace_{ace_index}_dry_target_temp")) diff --git a/custom_components/kobrax_lan/switch.py b/custom_components/kobrax_lan/switch.py index d0eafa2..5872e2a 100644 --- a/custom_components/kobrax_lan/switch.py +++ b/custom_components/kobrax_lan/switch.py @@ -85,14 +85,6 @@ class KobraXAceDryerSwitch(KobraXEntity, SwitchEntity): except KobraXApiError as err: raise ServiceValidationError(str(err)) from err - async def async_turn_off(self, **kwargs) -> None: - api = self.hass.data[DOMAIN][self._entry.entry_id]["api"] - try: - await api.async_set_ace_auto_feed(self._ace_id, False) - await self.coordinator.async_request_refresh() - except KobraXApiError as err: - raise ServiceValidationError(str(err)) from err - async def async_setup_entry(hass, entry, async_add_entities): coordinator = hass.data[DOMAIN][entry.entry_id]["coordinator"]