"""Print the canonical Nomen-AI training configuration."""
from pathlib import Path
import yaml


def main():
    cfg_path = Path(__file__).resolve().parents[1] / 'train_config.yaml'
    with cfg_path.open() as f:
        cfg = yaml.safe_load(f)
    print('Project:', cfg['project'])
    print('Base model:', cfg['base_model'])
    print('SFT dataset:', cfg['artifacts']['sft_dataset'])
    print('DPO dataset:', cfg['artifacts']['dpo_dataset'])
    print('SFT output:', cfg['artifacts']['sft_adapter'])
    print('DPO output:', cfg['artifacts']['dpo_adapter'])
    print('SFT effective batch:', cfg['sft']['effective_batch_size'])
    print('DPO effective batch:', cfg['dpo']['effective_batch_size'])
    print('Implemented roots:', cfg['controls']['implemented_roots'])


if __name__ == '__main__':
    main()