Spaces:
Sleeping
Sleeping
File size: 1,140 Bytes
fa7be76 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 |
import hydra
from hydra.core.hydra_config import HydraConfig
from omegaconf import DictConfig
def test_train_config(cfg_train: DictConfig) -> None:
"""Tests the training configuration provided by the `cfg_train` pytest fixture.
:param cfg_train: A DictConfig containing a valid training configuration.
"""
assert cfg_train
assert cfg_train.data
assert cfg_train.model
assert cfg_train.trainer
HydraConfig().set_config(cfg_train)
hydra.utils.instantiate(cfg_train.data)
hydra.utils.instantiate(cfg_train.model)
hydra.utils.instantiate(cfg_train.trainer)
def test_eval_config(cfg_eval: DictConfig) -> None:
"""Tests the evaluation configuration provided by the `cfg_eval` pytest fixture.
:param cfg_train: A DictConfig containing a valid evaluation configuration.
"""
assert cfg_eval
assert cfg_eval.data
assert cfg_eval.model
assert cfg_eval.trainer
HydraConfig().set_config(cfg_eval)
hydra.utils.instantiate(cfg_eval.data)
hydra.utils.instantiate(cfg_eval.model)
hydra.utils.instantiate(cfg_eval.trainer)
|