"""Train the local model from scratch on virtual FV3GFS-contract data."""

import argparse
from local_pipeline import load_config, train


def main() -> None:
    parser = argparse.ArgumentParser(description=__doc__)
    parser.add_argument("--config", default="conf/config.yaml")
    parser.add_argument(
        "--finetune",
        help="Optional checkpoint path. Without this argument, training starts from scratch.",
    )
    args = parser.parse_args()
    train(load_config(args.config), finetune=args.finetune)


if __name__ == "__main__":
    main()