fixing is_ptc_eligible

TheCGO · Jul 28, 2024 · e92e74f · e92e74f
1 parent 3976b40
commit e92e74f
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/fiscalsim_us/variables/gov/irs/credits/premium_tax_credit/is_ptc_eligible.py b/fiscalsim_us/variables/gov/irs/credits/premium_tax_credit/is_ptc_eligible.py
@@ -1,6 +1,6 @@
+import numpy as np
 from fiscalsim_us.model_api import *
 
-
 class is_ptc_eligible(Variable):
     value_type = bool
     entity = TaxUnit
@@ -15,4 +15,9 @@ def formula(tax_unit, period, parameters):
         on_marketplace = (
             add(tax_unit, period, ["has_marketplace_health_coverage"]) > 0
         )
-        return on_marketplace & eligibility.calc(income_level)
+        income_eligible = eligibility.calc(income_level)
+
+        on_marketplace = np.array(on_marketplace, dtype=bool)
+        income_eligible = np.array(income_eligible, dtype=bool)
+
+        return on_marketplace & income_eligible