Skip to content

Speed up ACS tax unit link inference#936

Merged
MaxGhenis merged 1 commit intomainfrom
codex/accelerate-acs-tax-unit-links
May 9, 2026
Merged

Speed up ACS tax unit link inference#936
MaxGhenis merged 1 commit intomainfrom
codex/accelerate-acs-tax-unit-links

Conversation

@MaxGhenis
Copy link
Copy Markdown
Contributor

Summary

  • Replace repeated per-household Pandas indexing in ACS spouse/parent inference with positional NumPy arrays
  • Keep the existing spouse, parent, and imputation rules unchanged

Tests

  • uv run ruff format --check policyengine_us_data/datasets/acs/acs_to_cps_columns.py
  • uv run ruff check policyengine_us_data/datasets/acs/acs_to_cps_columns.py
  • uv run python -m pytest tests/unit/datasets/test_acs_tax_unit_construction.py -q
  • git diff --check

@MaxGhenis MaxGhenis merged commit 15629bd into main May 9, 2026
12 checks passed
@MaxGhenis MaxGhenis deleted the codex/accelerate-acs-tax-unit-links branch May 9, 2026 21:08
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant