Skip to content

Conversation

@anth-volk
Copy link
Contributor

Fixes #213

Summary

  • Add DEFAULT_US_DATASET constant to explicitly define the default dataset for US simulations
  • The default is enhanced_cps_2024.h5 from HuggingFace (hf://policyengine/policyengine-us-data/enhanced_cps_2024.h5)
  • Update create_datasets(), load_datasets(), and ensure_datasets() functions to use the constant
  • Export the constant from the module for external use

Test plan

  • Verify DEFAULT_US_DATASET is accessible from policyengine.tax_benefit_models.us
  • Verify dataset functions default to Enhanced CPS when no dataset is specified

🤖 Generated with Claude Code

Add DEFAULT_US_DATASET constant to make the default dataset explicit
and easier to maintain. The default is enhanced_cps_2024.h5.

🤖 Generated with [Claude Code](https://claude.com/claude-code)

Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
@anth-volk
Copy link
Contributor Author

Closing in favor of #215 which targets the correct branch (0.x)

@anth-volk anth-volk closed this Jan 14, 2026
@anth-volk anth-volk deleted the use-enhanced-cps-default branch January 14, 2026 19:41
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.

Use Enhanced CPS by default for US simulations

2 participants