#!/usr/bin/env python3
# Author: Simon-Pierre Boucher — contact@spboucher.ai
"""Step 01 — Build the estimation sample.

Usage:  python scripts/01_build_sample.py
"""
import sys
from pathlib import Path

sys.path.insert(0, str(Path(__file__).resolve().parents[1] / "src"))

from wp11 import sample  # noqa: E402


def main() -> None:
    s = sample.build_and_save()
    print(f"\nEstimation sample written: {len(s):,} sales")
    print(f"  municipalities : {s['muni'].nunique():,}")
    print(f"  grid cells     : {s['grid'].nunique():,}")
    print(f"  grid5 cells    : {s['grid5'].nunique():,}")
    print(f"  classes        : {s.groupby('prop_class').size().to_dict()}")


if __name__ == "__main__":
    main()