Merge pull request #772 from PolicyEngine/enhanced-cps

Add support for enhanced CPS
PolicyEngine · Sep 22, 2023 · 00db175 · 00db175
2 parents 95b224a + 6d58a5d
commit 00db175
Show file tree

Hide file tree

Showing 4 changed files with 14 additions and 1 deletion.
diff --git a/changelog_entry.yaml b/changelog_entry.yaml
@@ -0,0 +1,4 @@
+- bump: minor
+  changes:
+    added:
+    - Support for the enhanced CPS in the US.
diff --git a/policyengine_api/country.py b/policyengine_api/country.py
@@ -79,6 +79,7 @@ def build_microsimulation_options(self) -> dict:
         elif self.country_id == "us":
             region = [
                 dict(name="us", label="the US"),
+                dict(name="enhanced_us", label="the US (enhanced CPS)"),
                 dict(name="al", label="Alabama"),
                 dict(name="ak", label="Alaska"),
                 dict(name="az", label="Arizona"),

diff --git a/policyengine_api/endpoints/economy/single_economy.py b/policyengine_api/endpoints/economy/single_economy.py
@@ -199,6 +199,14 @@ def compute_economy(
                     time_period,
                     original_household_weight * in_nyc,
                 )
+            elif region == "enhanced_us":
+                simulation: Microsimulation = (
+                    country.country_package.Microsimulation(
+                        reform=reform,
+                        dataset="enhanced_cps_2023",
+                    )
+                )
+                simulation.default_calculation_period = time_period
             else:
                 region_values = simulation.calculate("state_code_str").values
                 simulation.set_input(

diff --git a/setup.py b/setup.py
@@ -23,7 +23,7 @@
         "policyengine-ng==0.5.1",
         "policyengine-il==0.1.0",
         "policyengine_uk==0.55.2",
-        "policyengine_us==0.471.1",
+        "policyengine_us==0.472.0",
         "pymysql",
         "redis",
         "rq",