trying again

TheCGO · Jul 28, 2024 · e4ee6e0 · e4ee6e0
1 parent c8d9e84
commit e4ee6e0
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 or NaN values
+        ssi_eligible = np.nan_to_num(ssi_eligible, nan=False)
+        is_disabled = np.nan_to_num(is_disabled, nan=False)
+        is_blind = np.nan_to_num(is_blind, nan=False)
+
+        disabled_or_blind = is_disabled | is_blind
         in_age_range = p.age_range.calc(age)
+
+        # Handle potential None or NaN in in_age_range
+        in_age_range = np.nan_to_num(in_age_range, nan=False)
+
         return disabled_or_blind & ssi_eligible & in_age_range