Allow data-backed SSI criteria export

Author: MaxGhenis

policyengine_us_data/utils/dataset_validation.py

@@ -34,6 +34,19 @@
     }
 )
 
+# These variables have fallback formulas in policyengine-us for hand-authored
+# simulations, but enhanced datasets intentionally export stronger source-data
+# inputs that should override the fallback.
+DATA_OVERRIDABLE_COMPUTED_EXPORT_VARIABLES = frozenset(
+    {
+        "meets_ssi_disability_criteria",
+    }
+)
+
+ALLOWED_COMPUTED_EXPORT_VARIABLES = (
+    STRUCTURAL_COMPUTED_EXPORT_VARIABLES | DATA_OVERRIDABLE_COMPUTED_EXPORT_VARIABLES
+)
+
 AUXILIARY_ENTITY_PREFIXES = {
     "person_": "person",
     "tax_unit_": "tax_unit",
@@ -101,7 +114,7 @@ def computed_policyengine_us_variables_for_period(
 
     computed = set()
     for variable_name in variable_names:
-        if variable_name in STRUCTURAL_COMPUTED_EXPORT_VARIABLES:
+        if variable_name in ALLOWED_COMPUTED_EXPORT_VARIABLES:
             continue
         variable = tax_benefit_system.variables.get(variable_name)
         if variable is None:

tests/unit/test_extended_cps.py

@@ -248,6 +248,13 @@ def test_final_export_contract_rejects_computed_ss_total(self):
         with pytest.raises(DatasetContractError, match="social_security"):
             ExtendedCPS._assert_no_computed_variables_exported(data, 2024)
 
+    def test_final_export_contract_allows_data_overridden_ssi_disability_criteria(
+        self,
+    ):
+        data = {"meets_ssi_disability_criteria": {2024: np.array([True, False])}}
+
+        ExtendedCPS._assert_no_computed_variables_exported(data, 2024)
+
     def test_rename_imputed_to_inputs_maps_medicare_enrollment_to_take_up_input(self):
         data = {"medicare_enrolled": {2024: np.array([True, False])}}