diff --git a/tests/integration/stretched_config.yaml b/tests/integration/stretched_config.yaml index 2020c1f1..ec408c7a 100644 --- a/tests/integration/stretched_config.yaml +++ b/tests/integration/stretched_config.yaml @@ -8,7 +8,7 @@ defaults: - training: default - _self_ -no_validation: False +no_validation: True dataloader: @@ -22,8 +22,8 @@ dataloader: num_workers: training: 1 limit_batches: - training: 100 - validation: 100 + training: 8 + validation: 8 training: start: "2017-01-01" end: "2017-01-07" @@ -79,7 +79,6 @@ hardware: forcing_dataset: aifs-od-an-oper-0001-mars-o96-2016-2023-6h-v6 graph: test_stretched_graph.pt paths: - grids: null output: ${oc.env:PWD}/tmp_output/ # number of GPUs per node and number of nodes (for DDP) diff --git a/tests/integration/test_training_cycle.py b/tests/integration/test_training_cycle.py index e9b75183..c6829971 100644 --- a/tests/integration/test_training_cycle.py +++ b/tests/integration/test_training_cycle.py @@ -28,3 +28,12 @@ def test_training_cycle_architecture_configs(architecture_config) -> None: def test_training_cycle_grid_configs(stretched_config) -> None: AnemoiTrainer(stretched_config).train() shutil.rmtree(stretched_config.hardware.paths.output) + +if __name__ == "__main__": + from omegaconf import OmegaConf + from hydra import compose + from hydra import initialize + with initialize(version_base=None, config_path="", job_name="test_training"): + cfg = compose(config_name="stretched_config", overrides=[]) + OmegaConf.resolve(cfg) + test_training_cycle_architecture_configs(cfg)