INNER CODE UNIT · Python

handle_trigger_ml_training

3dg1luk43/ha_washdata · custom_components/ha_washdata/__init__.py:1234

        async def handle_trigger_ml_training(call: ServiceCall) -> None:
            device_id = _require_str(call.data.get("device_id"), "device_id")
            registry = dr.async_get(hass)
            device = registry.async_get(device_id)
            if not device:
                raise ValueError("Device not found")
            entry_id = next(iter(device.config_entries), None)
            if not entry_id or entry_id not in hass.data[DOMAIN]:
                raise ValueError("Integration not loaded")

            manager = hass.data[DOMAIN][entry_id]
            summary = await manager.async_run_ml_training(force=True)
            manager._logger.info("Manual ML training: %s", summary)

        hass.services.async_register(
            DOMAIN, SERVICE_TRIGGER_ML_TRAINING, handle_trigger_ml_training
        )

View source record →