Merge branch 'test/initial-integration-tests' of github.com:ecmwf/ane…

…moi-core into test/initial-integration-tests
ecmwf · Feb 25, 2025 · ff22e1f · ff22e1f
2 parents 36e788d + b4893e8
commit ff22e1f
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tests/integration/test_training_cycle.py b/tests/integration/test_training_cycle.py
@@ -29,10 +29,12 @@ 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
+    from omegaconf import OmegaConf
+
     with initialize(version_base=None, config_path="", job_name="test_training"):
         cfg = compose(config_name="stretched_config", overrides=[])
         OmegaConf.resolve(cfg)