confection_nested_config_with_registry_function_resolution.py

1from confection import Config, registry
2
3# 1. Define a registry to store functions
4@registry.optimizers.register("my_optimizer.v1")
5def create_optimizer(learn_rate: float, gamma: float):
6    return {"learn_rate": learn_rate, "gamma": gamma}
7
8# 2. Define the configuration
9config_str = """
10[training]
11patience = 10
12
13[training.optimizer]
14@optimizers = "my_optimizer.v1"
15learn_rate = 0.001
16gamma = 0.1
17"""
18
19# 3. Parse the config string
20config = Config().from_str(config_str)
21
22# 4. Resolve the configuration to create the objects
23resolved_config = registry.resolve(config)
24
25print(resolved_config["training"]["optimizer"])
26# Output: {'learn_rate': 0.001, 'gamma': 0.1}