Some checks failed
Code Quality Main / code-quality (push) Has been cancelled
Release Drafter / update_release_draft (push) Has been cancelled
Tests / run_tests_ubuntu (ubuntu-latest, 3.10) (push) Has been cancelled
Tests / run_tests_ubuntu (ubuntu-latest, 3.8) (push) Has been cancelled
Tests / run_tests_ubuntu (ubuntu-latest, 3.9) (push) Has been cancelled
Tests / run_tests_macos (macos-latest, 3.10) (push) Has been cancelled
Tests / run_tests_macos (macos-latest, 3.8) (push) Has been cancelled
Tests / run_tests_macos (macos-latest, 3.9) (push) Has been cancelled
Tests / run_tests_windows (windows-latest, 3.10) (push) Has been cancelled
Tests / run_tests_windows (windows-latest, 3.8) (push) Has been cancelled
Tests / run_tests_windows (windows-latest, 3.9) (push) Has been cancelled
Tests / code-coverage (push) Has been cancelled
38 lines
1.1 KiB
Python
38 lines
1.1 KiB
Python
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)
|