spb/wp11_uqo Public

1#!/usr/bin/env python32# Author: Simon-Pierre Boucher — contact@spboucher.ai3"""Step 01 — Build the estimation sample.45Usage:  python scripts/01_build_sample.py6"""7import sys8from pathlib import Path910sys.path.insert(0, str(Path(__file__).resolve().parents[1] / "src"))1112from wp11 import sample  # noqa: E402131415def main() -> None:16    s = sample.build_and_save()17    print(f"\nEstimation sample written: {len(s):,} sales")18    print(f"  municipalities : {s['muni'].nunique():,}")19    print(f"  grid cells     : {s['grid'].nunique():,}")20    print(f"  grid5 cells    : {s['grid5'].nunique():,}")21    print(f"  classes        : {s.groupby('prop_class').size().to_dict()}")222324if __name__ == "__main__":25    main()26