Add PUF to generated datasets

PolicyEngine · Sep 28, 2024 · b0d4711 · b0d4711
1 parent 91beb16
commit b0d4711
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 0 deletions.
diff --git a/Makefile b/Makefile
@@ -34,6 +34,8 @@ documentation:
 data:
 	python policyengine_us_data/datasets/acs/acs.py
 	python policyengine_us_data/datasets/cps/cps.py
+	python policyengine_us_data/datasets/puf/irs_puf.py
+	python policyengine_us_data/datasets/puf/puf.py
 	python policyengine_us_data/datasets/cps/extended_cps.py
 	python policyengine_us_data/datasets/cps/enhanced_cps.py
 

diff --git a/policyengine_us_data/datasets/puf/irs_puf.py b/policyengine_us_data/datasets/puf/irs_puf.py
@@ -42,3 +42,6 @@ class IRS_PUF_2015(IRS_PUF):
     puf_file_path = STORAGE_FOLDER / "puf_2015.csv"
     puf_demographics_file_path = STORAGE_FOLDER / "demographics_2015.csv"
     file_path = STORAGE_FOLDER / "irs_puf_2015.h5"
+
+if __name__ == "__main__":
+    IRS_PUF_2015().generate()
diff --git a/policyengine_us_data/datasets/puf/puf.py b/policyengine_us_data/datasets/puf/puf.py
@@ -514,3 +514,6 @@ class PUF_2024(PUF):
     "medicare_part_b_premiums": 0.137,
     "over_the_counter_health_expenses": 0.085,
 }
+
+if __name__ == "__main__":
+    PUF_2015().generate()