Skip to content
This repository was archived by the owner on Jun 14, 2026. It is now read-only.

Commit 0c019d9

Browse files
committed
Preserve filing status in PE rebuild profile
1 parent 9aae8de commit 0c019d9

3 files changed

Lines changed: 3 additions & 1 deletion

File tree

src/microplex_us/pipelines/pe_us_data_rebuild.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -76,6 +76,7 @@ def default_policyengine_us_data_rebuild_config(
7676
donor_imputer_backend="qrf",
7777
donor_imputer_condition_selection="pe_prespecified",
7878
donor_imputer_qrf_zero_threshold=0.05,
79+
donor_imputer_excluded_variables=(),
7980
prefer_cached_cps_asec_source=False,
8081
policyengine_direct_override_variables=(
8182
"health_savings_account_ald",

tests/pipelines/test_pe_us_data_rebuild.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -76,7 +76,7 @@ def test_default_policyengine_us_data_rebuild_config_uses_incumbent_defaults() -
7676
assert config.policyengine_calibration_min_active_households == 20
7777
assert config.donor_imputer_backend == "qrf"
7878
assert config.donor_imputer_condition_selection == "pe_prespecified"
79-
assert config.donor_imputer_excluded_variables == ("filing_status_code",)
79+
assert config.donor_imputer_excluded_variables == ()
8080
assert config.policyengine_direct_override_variables == (
8181
"health_savings_account_ald",
8282
"non_sch_d_capital_gains",

tests/pipelines/test_pe_us_data_rebuild_checkpoint.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -45,6 +45,7 @@ def test_default_policyengine_us_data_rebuild_checkpoint_config_sets_pe_context(
4545
assert config.policyengine_calibration_min_active_households == 20
4646
assert config.donor_imputer_backend == "qrf"
4747
assert config.donor_imputer_condition_selection == "pe_prespecified"
48+
assert config.donor_imputer_excluded_variables == ()
4849
assert config.policyengine_baseline_dataset == "/tmp/enhanced_cps_2024.h5"
4950
assert config.policyengine_targets_db == "/tmp/policy_data.db"
5051
assert config.policyengine_dataset_year == 2024

0 commit comments

Comments
 (0)