Skip to content

root

root

Root, experiment, and endpoint config registrations.

register

register(cs)

Store this domain's config nodes.

Parameters:

Name Type Description Default
cs ConfigStore

The global Hydra config store.

required
Source code in src/versatil/configs/store/root.py
def register(cs: ConfigStore) -> None:
    """Store this domain's config nodes.

    Args:
        cs: The global Hydra config store.
    """
    cs.store(name="config", node=MainConfig)
    cs.store(group="experiment", name="base", node=ExperimentConfig)
    cs.store(
        name="post_training_compression",
        node=PostTrainingCompressorConfig,
    )
    cs.store(
        name="explainability",
        node=ExplainabilityConfig,
    )
    cs.store(
        name="deployment",
        node=DeploymentConfig,
    )