OneScience-Group/MetNet-2

1#!/usr/bin/env python32import argparse3from pathlib import Path4import sys5 6sys.path.insert(0, str(Path(__file__).resolve().parents[1]))7from model.metnet_2 import load_config, write_fake_data8 9parser = argparse.ArgumentParser(description="Create selected MetNet-2 validation windows")10parser.add_argument("--config", default="conf/config.yaml")11parser.add_argument("--output", default=None)12args = parser.parse_args()13config = load_config(args.config)14output = args.output or config["data"]["path"]15print(write_fake_data(output, config["data"]["samples"], config["data"]["window"], config["data"]["halo"]))16