OneScience-Group/AtmosphericDA-DModel

1from pathlib import Path2import sys3 4ROOT = Path(__file__).resolve().parents[1]5sys.path.insert(0, str(ROOT))6 7from model.atmosphericda_dmodel import QGConfig, create_dataset, load_yaml8 9 10if __name__ == "__main__":11    config = load_yaml(ROOT / "conf/config.yaml")12    qg = QGConfig(**config["qg"])13    settings = config["data"]14    output = ROOT / config["paths"]["data"]15    create_dataset(str(output), settings["samples"], qg, settings["seed"],16                   settings["observations_per_batch"], settings["observation_variance"],17                   settings["analysis_gain"], settings["format_version"])18    print(f"saved {output}: state=[{settings['samples']},2,{qg.ny},{qg.nx}] observations=[{settings['samples']},12,50]")19