fix

PavelMakarchuk · PavelMakarchuk · commit 6755bbb32f76 · 2025-02-17T16:46:41.000+01:00
diff --git a/policyengine_us_data/datasets/cps/cps.py b/policyengine_us_data/datasets/cps/cps.py
@@ -263,7 +263,7 @@ def _assign_some_newborns_to_pregnancy(
         )
         return age
 
-    cps["age"] = _assign_some_newborns_to_pregnancy(cps["age"])
+    cps["age"] = _assign_some_newborns_to_pregnancy(cps["age"], person)
 
     def children_per_parent(col: str) -> pd.DataFrame:
         """Calculate number of children in the household using parental

Original file line number	Diff line number	Diff line change
`@@ -263,7 +263,7 @@ def _assign_some_newborns_to_pregnancy(`
`263`	`263`	`)`
`264`	`264`	`return age`
`265`	`265`
`266`		`- cps["age"] = _assign_some_newborns_to_pregnancy(cps["age"])`
	`266`	`+ cps["age"] = _assign_some_newborns_to_pregnancy(cps["age"], person)`
`267`	`267`
`268`	`268`	`def children_per_parent(col: str) -> pd.DataFrame:`
`269`	`269`	`"""Calculate number of children in the household using parental`