WIP stretched grid debugging

ecmwf · Feb 24, 2025 · 9d7dfbb · 9d7dfbb
1 parent 41a0cdd
commit 9d7dfbb
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 4 deletions.
diff --git 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
@@ -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)