def METHOD_NAME() -> argparse.ArgumentParser:
    parser = argparse.ArgumentParser()
    parser.add_argument("reservoir_model_path", type=str, help="Input Zarr path")
    parser.add_argument("output_path", type=str, help="Directory to save outputs to")
    parser.add_argument(
        "validation_config_path", type=str, help="Path to validation data config"
    )
    parser.add_argument(
        "n_synchronize",
        type=int,
        help=(
            "Number of timesteps from start of validation to use in reservoir "
            "synchronization (not used in prediction)."
        ),
    )
    parser.add_argument(
        "--n-validation-batches",
        type=int,
        default=None,
        help="Number of batch data netcdfs to use for validation. Defaults to use all.",
    )
    return parser