from pathlib import Path


def build_args(parser):
    """Constructs the command-line arguments for ``prepare_config``."""
    parser.add_argument("--in-template", type=Path, required=True)
    parser.add_argument("--out-config", type=Path, required=True)
    parser.add_argument("--expdir", type=Path, required=True)
    parser.add_argument("--dataset-path", type=Path, required=True)

    parser.add_argument("--num-classes", default=None, type=int)
    parser.add_argument("--num-workers", default=4, type=int)
    parser.add_argument("--batch-size", default=64, type=int)
    parser.add_argument("--max-image-size", default=256, type=int)

    return parser