handling None type for CO

TheCGO · Jul 28, 2024 · b622da7 · b622da7
1 parent 4002161
commit b622da7
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/fiscalsim_us/variables/gov/states/co/ssa/state_suplement/co_state_supplement_eligible.py b/fiscalsim_us/variables/gov/states/co/ssa/state_suplement/co_state_supplement_eligible.py
@@ -12,8 +12,18 @@ def formula(person, period, parameters):
         ssi_eligible = person("is_ssi_eligible_individual", period)
         is_disabled = person("is_ssi_disabled", period)
         is_blind = person("is_blind", period)
-        disabled_or_blind = is_disabled | is_blind
         age = person("age", period)
         p = parameters(period).gov.states.co.ssa.state_supplement
+
+        # Handle potential None values
+        ssi_eligible = ssi_eligible.fillna(False)
+        is_disabled = is_disabled.fillna(False)
+        is_blind = is_blind.fillna(False)
+
+        disabled_or_blind = is_disabled | is_blind
         in_age_range = p.age_range.calc(age)
+
+        # Handle potential None in in_age_range
+        in_age_range = in_age_range.fillna(False)
+
         return disabled_or_blind & ssi_eligible & in_age_range