| #!/usr/bin/env python3 | |
| import argparse | |
| from pathlib import Path | |
| import sys | |
| sys.path.insert(0, str(Path(__file__).resolve().parents[1])) | |
| from model.metnet_2 import load_config, write_fake_data | |
| parser = argparse.ArgumentParser(description="Create selected MetNet-2 validation windows") | |
| parser.add_argument("--config", default="conf/config.yaml") | |
| parser.add_argument("--output", default=None) | |
| args = parser.parse_args() | |
| config = load_config(args.config) | |
| output = args.output or config["data"]["path"] | |
| print(write_fake_data(output, config["data"]["samples"], config["data"]["window"], config["data"]["halo"])) | |