from pathlib import Path


def build_args(parser):
    """Constructs the command-line arguments for ``catalyst-dl init``."""
    parser.add_argument(
        "-p",
        "--pipeline",
        type=str,
        default=None,
        choices=["empty", "classification", "segmentation", "detection"],
        help="select a Catalyst pipeline",
    )
    parser.add_argument(
        "-i",
        "--interactive",
        action="store_true",
        help="use interactive wizard to setup Catalyst pipeline",
    )
    parser.add_argument(
        "-o", "--out-dir", type=Path, default="./", help="path where to init"
    )

    return parser