confection_config_registry_function_parse_and_resolve.py

1from confection import Config, registry
2
3# 1. Define a registry and register a function
4@registry.optimizers.register("my_cool_optimizer.v1")
5def make_optimizer(learn_rate: float):
6    return {"optimizer": "Adam", "lr": learn_rate}
7
8# 2. Define your configuration
9config_str = """
10[optimizer]
11@optimizers = "my_cool_optimizer.v1"
12learn_rate = 0.001
13"""
14
15# 3. Parse and resolve the configuration
16config = Config().from_str(config_str)
17resolved = registry.resolve(config)
18
19print(resolved["optimizer"])
20# {'optimizer': 'Adam', 'lr': 0.001}