Fix SSI imputation PR checks

MaxGhenis · MaxGhenis · commit b090a9fe610b · 2026-05-23T15:38:00.000-04:00
diff --git a/changelog.d/1123.changed.md b/changelog.d/1123.changed.md
@@ -0,0 +1 @@
+Improve SSI disability imputation by using comparable CPS and SIPP difficulty flags and refreshing CPS-only disability attributes on PUF clones.
diff --git a/policyengine_us_data/datasets/sipp/sipp.py b/policyengine_us_data/datasets/sipp/sipp.py
@@ -531,6 +531,8 @@ def build_ssi_disability_training_frame(
     social_security_amount = (
         df["TSSSAMT"] if "TSSSAMT" in df else pd.Series(0.0, index=df.index)
     )
+    # SSDI receipt is evidence for disability status, but the training label
+    # below remains SSI receipt reason, not a broad SSA disability determination.
     df["social_security_disability"] = np.where(
         _yes(df, "ESSRSN2YN"),
         social_security_amount.fillna(0).astype(float) * 12,
diff --git a/pyproject.toml b/pyproject.toml
@@ -22,7 +22,7 @@ classifiers = [
     "Programming Language :: Python :: 3.14",
 ]
 dependencies = [
-    "policyengine-us==1.705.1",
+    "policyengine-us==1.705.6",
     # policyengine-core 3.26.1 is the current 3.26.x runtime and includes the fix for
     # PolicyEngine/policyengine-core#482 (user-set ETERNITY inputs lost
     # after _invalidate_all_caches) and is required by policyengine-us 1.682.1+.
diff --git a/uv.lock b/uv.lock

Original file line number	Diff line number	Diff line change
`@@ -0,0 +1 @@`
	`1`	`+Improve SSI disability imputation by using comparable CPS and SIPP difficulty flags and refreshing CPS-only disability attributes on PUF clones.`
Original file line number	Diff line number	Diff line change
`@@ -531,6 +531,8 @@ def build_ssi_disability_training_frame(`
`531`	`531`	`social_security_amount = (`
`532`	`532`	`df["TSSSAMT"] if "TSSSAMT" in df else pd.Series(0.0, index=df.index)`
`533`	`533`	`)`
	`534`	`+ # SSDI receipt is evidence for disability status, but the training label`
	`535`	`+ # below remains SSI receipt reason, not a broad SSA disability determination.`
`534`	`536`	`df["social_security_disability"] = np.where(`
`535`	`537`	`_yes(df, "ESSRSN2YN"),`
`536`	`538`	`social_security_amount.fillna(0).astype(float) * 12,`
Original file line number	Diff line number	Diff line change
`@@ -22,7 +22,7 @@ classifiers = [`
`22`	`22`	`"Programming Language :: Python :: 3.14",`
`23`	`23`	`]`
`24`	`24`	`dependencies = [`
`25`		`- "policyengine-us==1.705.1",`
	`25`	`+ "policyengine-us==1.705.6",`
`26`	`26`	`# policyengine-core 3.26.1 is the current 3.26.x runtime and includes the fix for`
`27`	`27`	`# PolicyEngine/policyengine-core#482 (user-set ETERNITY inputs lost`
`28`	`28`	`# after _invalidate_all_caches) and is required by policyengine-us 1.682.1+.`