Add PUF to generated datasets

nikhilwoodruff · nikhilwoodruff · commit b0d4711794c2 · 2024-09-28T14:08:28.000+01:00
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()

Original file line number	Diff line number	Diff line change
`@@ -514,3 +514,6 @@ class PUF_2024(PUF):`
`514`	`514`	`"medicare_part_b_premiums": 0.137,`
`515`	`515`	`"over_the_counter_health_expenses": 0.085,`
`516`	`516`	`}`
	`517`	`+`
	`518`	`+if __name__ == "__main__":`
	`519`	`+ PUF_2015().generate()`