scripts/download_kaggle_data.py

from __future__ import annotations

import argparse
import subprocess
from pathlib import Path


COMPETITION = "house-prices-advanced-regression-techniques"


def main() -> None:
    parser = argparse.ArgumentParser(description="Download Kaggle House Prices data.")
    parser.add_argument("--output", type=Path, default=Path("data/raw"))
    args = parser.parse_args()
    args.output.mkdir(parents=True, exist_ok=True)
    subprocess.run(
        [
            "kaggle",
            "competitions",
            "download",
            "-c",
            COMPETITION,
            "-p",
            str(args.output),
            "--unzip",
        ],
        check=True,
    )


if __name__ == "__main__":
    main()