diff --git a/README.md b/README.md index 513b7a0..28e407b 100644 --- a/README.md +++ b/README.md @@ -116,3 +116,4 @@ class CustomDataset: ### ACS - OpenFisca-US-compatible - Contains OpenFisca-US-compatible input arrays. +### hello world hello \ No newline at end of file diff --git a/openfisca_us_data/datasets/cps/cps.py b/openfisca_us_data/datasets/cps/cps.py index a5e9982..c3c9c5a 100644 --- a/openfisca_us_data/datasets/cps/cps.py +++ b/openfisca_us_data/datasets/cps/cps.py @@ -40,7 +40,7 @@ def generate(year: int) -> None: add_personal_variables(cps, person) add_personal_income_variables(cps, person) add_SPM_variables(cps, spm_unit) - + add_tax_unit_variables(cps, tax_unit) raw_data.close() cps.close() @@ -147,3 +147,11 @@ def add_SPM_variables(cps: h5py.File, spm_unit: DataFrame): for openfisca_variable, asec_variable in SPM_RENAMES.items(): cps[openfisca_variable] = spm_unit[asec_variable] + + +def add_tax_unit_variables(cps: h5py.File, tax_unit: DataFrame): + cps["e00200p"] = tax_unit.ACTC_CRD + cps["e00200s"] = tax_unit.AGI + cps["e00900s"] = tax_unit.CTC_CRD + cps["pencon_p"] = tax_unit.FED_RET + cps["pencon_s"] = tax_unit.MARG_TAX