Hydra Compose API Unit Tests Example

hydra_compose_api_unit_test_ex.py

@pytest.mark.parametrize("network_size", ["small", "big", "huge"])
def test_train_network_size(monkeypatch, tmp_path, capsys, network_size):
    with initialize(config_path="../leela_zero_pytorch/conf"):
        cfg = compose(
            config_name="config",
            overrides=[
                f"+network={network_size}",
                "data.train_data_dir=tests/test-data",
                "data.train_dataloader_conf.batch_size=2",
                "data.val_data_dir=tests/test-data",
                "data.val_dataloader_conf.batch_size=2",
                "data.test_data_dir=tests/test-data",
                "data.test_dataloader_conf.batch_size=2",
                f"+pl_trainer.default_root_dir={tmp_path}",
                "+pl_trainer.fast_dev_run=true",
                "~logger",
            ],
        )