nat.plugins.config_optimizer.register#

Registry for optimizer strategies (numeric/parameter and GA prompt).

Classes#

_ParameterOptimizerRunner

Runner that delegates to optimize_parameters (sync) via asyncio.to_thread.

Functions#

_parameter_optimizer_build(...)

register_numeric_optimizer(config)

_ga_prompt_optimizer_build(...)

register_ga_prompt_optimizer(config)

Module Contents#

class _ParameterOptimizerRunner#

Bases: nat.plugins.config_optimizer.parameters.base.BaseParameterOptimizer

Runner that delegates to optimize_parameters (sync) via asyncio.to_thread.

async run(
*,
base_cfg: nat.data_models.config.Config,
full_space: dict[str, nat.data_models.optimizable.SearchSpace],
optimizer_config: nat.data_models.optimizer.OptimizerConfig,
opt_run_config: nat.data_models.optimizer.OptimizerRunConfig,
callback_manager=None,
) → tuple[nat.data_models.config.Config, dict[str, object], int]#
async _parameter_optimizer_build(
_config: nat.data_models.optimizer.OptunaParameterOptimizationConfig,
) → collections.abc.AsyncIterator[_ParameterOptimizerRunner]#
async register_numeric_optimizer(
config: nat.data_models.optimizer.OptunaParameterOptimizationConfig,
)#
async _ga_prompt_optimizer_build(
_config: nat.data_models.optimizer.GAPromptOptimizationConfig,
) → collections.abc.AsyncIterator[nat.plugins.config_optimizer.prompts.ga_prompt_optimizer.GAPromptOptimizer]#
async register_ga_prompt_optimizer(
config: nat.data_models.optimizer.GAPromptOptimizationConfig,
)#