|
34 | 34 | "docstring": "Impute rent and real_estate_taxes from ACS with state.\n\nArgs:\n data: CPS data dict.\n state_fips: State FIPS per household.\n time_period: Tax year.\n dataset_path: Path to CPS h5 for Microsimulation.\n\nReturns:\n Updated data dict.", |
35 | 35 | "id": "acs_qrf", |
36 | 36 | "kind": "function", |
37 | | - "line": 334, |
| 37 | + "line": 485, |
38 | 38 | "metadata": { |
39 | 39 | "api_refs": [ |
40 | 40 | "policyengine_us_data.calibration.source_impute._impute_acs" |
|
61 | 61 | "docstring": "\"Add auto loan balance, interest and net_worth variable.", |
62 | 62 | "id": "add_auto_loan", |
63 | 63 | "kind": "function", |
64 | | - "line": 2819, |
| 64 | + "line": 2859, |
65 | 65 | "metadata": { |
66 | 66 | "api_refs": [ |
67 | 67 | "policyengine_us_data.datasets.cps.cps.add_auto_loan_interest_and_net_worth" |
|
142 | 142 | "docstring": "Impute ORG-derived wage and union inputs onto CPS persons.", |
143 | 143 | "id": "add_org_inputs", |
144 | 144 | "kind": "function", |
145 | | - "line": 2703, |
| 145 | + "line": 2743, |
146 | 146 | "metadata": { |
147 | 147 | "api_refs": [ |
148 | 148 | "policyengine_us_data.datasets.cps.cps.add_org_labor_market_inputs" |
|
572 | 572 | "docstring": "Build sparse calibration matrix.\n\nTwo-phase build: (1) simulate each clone and save\nCOO entries to disk, (2) assemble CSR from caches.\n\nArgs:\n geography: GeographyAssignment with state_fips,\n cd_geoid, block_geoid arrays and n_records,\n n_clones attributes.\n sim: Microsimulation for parameters and entity\n relationships.\n target_filter: Dict for target_overview filtering.\n hierarchical_domains: Domain names for\n hierarchical uprating + CD reconciliation.\n cache_dir: Directory for per-clone COO caches.\n If None, COO data held in memory.\n sim_modifier: Optional callback(sim, clone_idx)\n called per clone after state_fips is set but\n before cache clearing. Use for takeup\n re-randomization.\n rerandomize_takeup: If True, use geo-salted\n entity-level takeup draws instead of base h5\n takeup values for takeup-affected targets.\n county_level: If True (default), iterate counties\n within each state during precomputation. If\n False, compute once per state and alias to all\n counties (faster for county-invariant vars).\n\nReturns:\n (targets_df, X_sparse, target_names)", |
573 | 573 | "id": "build_matrix", |
574 | 574 | "kind": "function", |
575 | | - "line": 2549, |
| 575 | + "line": 2567, |
576 | 576 | "metadata": { |
577 | 577 | "api_refs": [ |
578 | 578 | "policyengine_us_data.calibration.unified_matrix_builder.UnifiedMatrixBuilder.build_matrix" |
|
604 | 604 | "docstring": "Build a sparse matrix by materializing mixed-geography chunks.\n\nThin facade: target querying, uprating, constraint extraction,\nand manifest handling live here; chunking, per-chunk execution,\nand streaming final assembly live in\n:class:`ChunkedMatrixAssembler`.", |
605 | 605 | "id": "build_matrix_chunked", |
606 | 606 | "kind": "function", |
607 | | - "line": 3314, |
| 607 | + "line": 3332, |
608 | 608 | "metadata": { |
609 | 609 | "api_refs": [ |
610 | 610 | "policyengine_us_data.calibration.unified_matrix_builder.UnifiedMatrixBuilder.build_matrix_chunked" |
|
3461 | 3461 | "docstring": "Impute net_worth and auto_loan from SCF.\n\nArgs:\n data: CPS data dict.\n state_fips: State FIPS per household.\n time_period: Tax year.\n dataset_path: Path to CPS h5 for Microsimulation.\n\nReturns:\n Updated data dict.", |
3462 | 3462 | "id": "scf_qrf", |
3463 | 3463 | "kind": "function", |
3464 | | - "line": 819, |
| 3464 | + "line": 922, |
3465 | 3465 | "metadata": { |
3466 | 3466 | "api_refs": [ |
3467 | 3467 | "policyengine_us_data.calibration.source_impute._impute_scf" |
|
3515 | 3515 | "docstring": "Impute tip_income, liquid assets, and vehicle signals from SIPP.\n\nArgs:\n data: CPS data dict.\n state_fips: State FIPS per household.\n time_period: Tax year.\n dataset_path: Path to CPS h5 for Microsimulation.\n\nReturns:\n Updated data dict.", |
3516 | 3516 | "id": "sipp_qrf", |
3517 | 3517 | "kind": "function", |
3518 | | - "line": 435, |
| 3518 | + "line": 586, |
3519 | 3519 | "metadata": { |
3520 | 3520 | "api_refs": [ |
3521 | 3521 | "policyengine_us_data.calibration.source_impute._impute_sipp" |
|
3542 | 3542 | "docstring": "Re-impute ACS/SIPP/ORG/SCF variables from donor surveys.\n\nOverwrites existing imputed values in data. ACS uses\nstate_fips as a QRF predictor; ORG uses state plus labor-market\npredictors; SIPP and SCF use only demographic and financial\npredictors (no state data).\n\nArgs:\n data: CPS dataset dict {variable: {time_period: array}}.\n state_fips: State FIPS per household.\n time_period: Tax year.\n dataset_path: Path to CPS h5 for Microsimulation.\n skip_acs: Skip ACS imputation.\n skip_sipp: Skip SIPP imputation.\n skip_org: Skip ORG imputation.\n skip_scf: Skip SCF imputation.\n\nReturns:\n Updated data dict with re-imputed variables.", |
3543 | 3543 | "id": "source_impute", |
3544 | 3544 | "kind": "function", |
3545 | | - "line": 187, |
| 3545 | + "line": 180, |
3546 | 3546 | "metadata": { |
3547 | 3547 | "api_refs": [ |
3548 | 3548 | "policyengine_us_data.calibration.source_impute.impute_source_variables" |
|
0 commit comments