policyengine-us 1.455.0__py3-none-any.whl → 1.503.1__py3-none-any.whl

policyengine_us/variables/gov/states/ga/dfcs/tanf/eligibility/ga_tanf_eligible.py

@@ -0,0 +1,43 @@
+from policyengine_us.model_api import *
+
+
+class ga_tanf_eligible(Variable):
+    value_type = bool
+    entity = SPMUnit
+    label = (
+        "Eligible for Georgia Temporary Assistance for Needy Families (TANF)"
+    )
+    definition_period = MONTH
+    reference = (
+        "https://rules.sos.ga.gov/gac/290-2-28",
+        "https://dfcs.georgia.gov/services/temporary-assistance-needy-families/tanf-eligibility-requirements",
+        "https://pamms.dhs.ga.gov/dfcs/tanf/",
+    )
+    defined_for = StateCode.GA
+
+    def formula(spm_unit, period, parameters):
+        person = spm_unit.members
+
+        # Must meet demographic requirements (minor child OR pregnant woman)
+        # Use federal demographic eligibility directly
+        demographic_eligible = spm_unit("is_demographic_tanf_eligible", period)
+
+        # Must have at least one citizen or legal immigrant
+        # Use federal immigration eligibility
+        immigration_status_eligible = spm_unit.any(
+            person("is_citizen_or_legal_immigrant", period)
+        )
+
+        # Must meet income requirements (gross and net income tests)
+        income_eligible = spm_unit("ga_tanf_income_eligible", period)
+
+        # Must meet resource requirements (assets <= $1,000)
+        resources_eligible = spm_unit("ga_tanf_resources_eligible", period)
+
+        # All requirements must be met
+        return (
+            demographic_eligible
+            & immigration_status_eligible
+            & income_eligible
+            & resources_eligible
+        )

policyengine_us/variables/gov/states/ga/dfcs/tanf/eligibility/ga_tanf_income_eligible.py

@@ -0,0 +1,31 @@
+from policyengine_us.model_api import *
+
+
+class ga_tanf_income_eligible(Variable):
+    value_type = bool
+    entity = SPMUnit
+    label = "Eligible for Georgia TANF due to income"
+    definition_period = MONTH
+    reference = (
+        "https://pamms.dhs.ga.gov/dfcs/tanf/1525/",
+        "https://pamms.dhs.ga.gov/dfcs/tanf/appendix-a/",
+    )
+    defined_for = StateCode.GA
+
+    def formula(spm_unit, period, parameters):
+        # Gross income test - use federal TANF variables
+        gross_income = add(
+            spm_unit,
+            period,
+            ["tanf_gross_earned_income", "tanf_gross_unearned_income"],
+        )
+
+        gross_income_ceiling = spm_unit("ga_tanf_gross_income_ceiling", period)
+        passes_gross_test = gross_income <= gross_income_ceiling
+
+        # Net income test
+        countable_income = spm_unit("ga_tanf_countable_income", period)
+        standard_of_need = spm_unit("ga_tanf_standard_of_need", period)
+        passes_net_test = countable_income < standard_of_need
+
+        return passes_gross_test & passes_net_test

policyengine_us/variables/gov/states/ga/dfcs/tanf/eligibility/ga_tanf_resources_eligible.py

@@ -0,0 +1,22 @@
+from policyengine_us.model_api import *
+
+
+class ga_tanf_resources_eligible(Variable):
+    value_type = bool
+    entity = SPMUnit
+    label = "Eligible for Georgia TANF due to resources"
+    definition_period = MONTH
+    reference = (
+        "https://rules.sos.ga.gov/gac/290-2-28-.13",
+        "https://pamms.dhs.ga.gov/dfcs/tanf/appendix-a/",
+    )
+    defined_for = StateCode.GA
+
+    def formula(spm_unit, period, parameters):
+        p = parameters(period).gov.states.ga.dfcs.tanf.resources
+        # For simplified implementation, use SPM unit cash assets directly
+        # In a more complete implementation, this would account for
+        # excluded resources (primary residence, household goods, etc.)
+        # and vehicle value limits
+        cash_assets = spm_unit("spm_unit_cash_assets", period.this_year)
+        return cash_assets <= p.limit

policyengine_us/variables/gov/states/ga/dfcs/tanf/ga_tanf.py

@@ -0,0 +1,23 @@
+from policyengine_us.model_api import *
+
+
+class ga_tanf(Variable):
+    value_type = float
+    entity = SPMUnit
+    label = "Georgia Temporary Assistance for Needy Families (TANF)"
+    unit = USD
+    definition_period = MONTH
+    reference = (
+        "https://rules.sos.ga.gov/gac/290-2-28",
+        "https://pamms.dhs.ga.gov/dfcs/tanf/",
+    )
+    defined_for = "ga_tanf_eligible"
+
+    def formula(spm_unit, period, parameters):
+        standard_of_need = spm_unit("ga_tanf_standard_of_need", period)
+        countable_income = spm_unit("ga_tanf_countable_income", period)
+        family_maximum = spm_unit("ga_tanf_family_maximum", period)
+
+        # Benefit = MIN(Standard of Need - Countable Income, Family Maximum)
+        benefit_before_max = max_(standard_of_need - countable_income, 0)
+        return min_(benefit_before_max, family_maximum)

policyengine_us/variables/gov/states/ga/dfcs/tanf/income/deductions/ga_tanf_childcare_deduction.py

@@ -0,0 +1,31 @@
+from policyengine_us.model_api import *
+
+
+class ga_tanf_childcare_deduction(Variable):
+    value_type = float
+    entity = SPMUnit
+    label = "Georgia TANF childcare deduction"
+    unit = USD
+    definition_period = MONTH
+    reference = (
+        "https://pamms.dhs.ga.gov/dfcs/tanf/1615/",
+        "https://pamms.dhs.ga.gov/dfcs/tanf/1605/",
+    )
+    defined_for = StateCode.GA
+
+    def formula(spm_unit, period, parameters):
+        p = parameters(period).gov.states.ga.dfcs.tanf.income.deductions
+        person = spm_unit.members
+        dependent = person("is_tax_unit_dependent", period)
+        age = person("monthly_age", period)
+        childcare_expenses = spm_unit("childcare_expenses", period)
+
+        # PAMMS 1615: "$200 monthly for each child under the age of two"
+        # and "$175 monthly for each individual age two or above"
+        # PAMMS 1605: Childcare is an earned income deduction applied
+        # in order after work expense
+        # Calculate max deduction per dependent based on age
+        childcare_deduction_person = p.childcare.calc(age) * dependent
+        total_childcare_deduction = spm_unit.sum(childcare_deduction_person)
+
+        return min_(childcare_expenses, total_childcare_deduction)

policyengine_us/variables/gov/states/ga/dfcs/tanf/income/ga_tanf_countable_income.py

@@ -0,0 +1,38 @@
+from policyengine_us.model_api import *
+
+
+class ga_tanf_countable_income(Variable):
+    value_type = float
+    entity = SPMUnit
+    label = "Georgia TANF countable income"
+    unit = USD
+    definition_period = MONTH
+    reference = (
+        "https://pamms.dhs.ga.gov/dfcs/tanf/1605/",
+        "https://pamms.dhs.ga.gov/dfcs/tanf/1615/",
+    )
+    defined_for = StateCode.GA
+
+    def formula(spm_unit, period, parameters):
+        # PAMMS 1605 Step 8: "Allow all applicable earned income deductions
+        # to the gross countable earned income of each employed individual
+        # to determine the net earned income"
+        # PAMMS 1615: "Deductions are applied in the order listed:
+        # 1. $250 standard work expense, 2. Dependent care"
+
+        # Sum person-level earned income after work expense disregard
+        earned_after_disregard = add(
+            spm_unit, period, ["ga_tanf_earned_income_after_disregard"]
+        )
+
+        # Apply childcare deduction to earned income only
+        childcare_deduction = spm_unit("ga_tanf_childcare_deduction", period)
+        net_earned = max_(earned_after_disregard - childcare_deduction, 0)
+
+        # PAMMS 1605 Step 9: "Add the unearned income of all individuals
+        # to the net earned income to determine the net countable income"
+        # PAMMS 1605: "Deductions are not allowed to unearned income"
+        unearned = add(spm_unit, period, ["tanf_gross_unearned_income"])
+
+        # Net countable income = net earned + unearned
+        return net_earned + unearned

policyengine_us/variables/gov/states/ga/dfcs/tanf/income/ga_tanf_earned_income_after_disregard.py

@@ -0,0 +1,23 @@
+from policyengine_us.model_api import *
+
+
+class ga_tanf_earned_income_after_disregard(Variable):
+    value_type = float
+    entity = Person
+    label = "Georgia TANF earned income after work expense disregard"
+    unit = USD
+    definition_period = MONTH
+    reference = (
+        "https://pamms.dhs.ga.gov/dfcs/tanf/1615/",
+        "https://pamms.dhs.ga.gov/dfcs/tanf/1605/",
+    )
+    defined_for = StateCode.GA
+
+    def formula(person, period, parameters):
+        p = parameters(period).gov.states.ga.dfcs.tanf.income.deductions
+        # PAMMS 1615: "$250 is subtracted from the earned income of
+        # each employed individual"
+        # PAMMS 1605 Step 8: Apply earned income deductions to
+        # "each employed individual"
+        gross_earned = person("tanf_gross_earned_income", period)
+        return max_(gross_earned - p.work_expense, 0)

policyengine_us/variables/gov/states/hi/dhs/tanf/eligibility/hi_tanf_eligible.py

@@ -0,0 +1,25 @@
+from policyengine_us.model_api import *
+
+
+class hi_tanf_eligible(Variable):
+    value_type = bool
+    entity = SPMUnit
+    label = "Eligible for Hawaii TANF"
+    definition_period = MONTH
+    reference = (
+        "https://humanservices.hawaii.gov/wp-content/uploads/2024/12/Hawaii_TANF_State_Plan_Signed_Certified-Eff_20231001.pdf#page=6",
+    )
+    defined_for = StateCode.HI
+
+    def formula(spm_unit, period, parameters):
+        demographic_eligible = spm_unit("is_demographic_tanf_eligible", period)
+        immigration_eligible = (
+            add(spm_unit, period, ["is_citizen_or_legal_immigrant"]) > 0
+        )
+        gross_income_eligible = spm_unit(
+            "hi_tanf_gross_income_eligible", period
+        )
+        # NOTE: Hawaii disregards assets since April 18, 2013
+        return (
+            demographic_eligible & immigration_eligible & gross_income_eligible
+        )

policyengine_us/variables/gov/states/hi/dhs/tanf/eligibility/hi_tanf_gross_income_eligible.py

@@ -0,0 +1,27 @@
+from policyengine_us.model_api import *
+
+
+class hi_tanf_gross_income_eligible(Variable):
+    value_type = bool
+    entity = SPMUnit
+    label = "Hawaii TANF gross income eligibility"
+    definition_period = MONTH
+    reference = (
+        "https://humanservices.hawaii.gov/wp-content/uploads/2024/12/Hawaii_TANF_State_Plan_Signed_Certified-Eff_20231001.pdf#page=19",
+    )
+    defined_for = StateCode.HI
+
+    def formula(spm_unit, period, parameters):
+        p = parameters(period).gov.states.hi.dhs.tanf
+
+        gross_income = spm_unit("hi_tanf_gross_income", period)
+
+        # Gross income limit = 185% of Standard of Need
+        unit_size = spm_unit("spm_unit_size", period)
+        capped_size = min_(unit_size, p.max_unit_size)
+        standard_of_need = p.standard_of_need.amount[capped_size]
+        gross_income_limit = (
+            standard_of_need * p.eligibility.gross_income_limit_rate
+        )
+
+        return gross_income <= gross_income_limit

policyengine_us/variables/gov/states/hi/dhs/tanf/hi_tanf.py

@@ -0,0 +1,20 @@
+from policyengine_us.model_api import *
+
+
+class hi_tanf(Variable):
+    value_type = float
+    entity = SPMUnit
+    label = "Hawaii TANF benefit amount"
+    unit = USD
+    definition_period = MONTH
+    reference = (
+        "https://humanservices.hawaii.gov/wp-content/uploads/2024/12/Hawaii_TANF_State_Plan_Signed_Certified-Eff_20231001.pdf#page=22",
+    )
+    defined_for = "hi_tanf_eligible"
+
+    def formula(spm_unit, period, parameters):
+        maximum_benefit = spm_unit("hi_tanf_maximum_benefit", period)
+        countable_income = spm_unit("hi_tanf_countable_income", period)
+
+        # Benefit = SOA - countable income, floored at 0
+        return max_(maximum_benefit - countable_income, 0)

policyengine_us/variables/gov/states/hi/dhs/tanf/hi_tanf_maximum_benefit.py

@@ -0,0 +1,29 @@
+from policyengine_us.model_api import *
+
+
+class hi_tanf_maximum_benefit(Variable):
+    value_type = float
+    entity = SPMUnit
+    label = "Hawaii TANF maximum benefit"
+    unit = USD
+    definition_period = MONTH
+    reference = (
+        "https://humanservices.hawaii.gov/wp-content/uploads/2024/12/Hawaii_TANF_State_Plan_Signed_Certified-Eff_20231001.pdf#page=22",
+    )
+    defined_for = StateCode.HI
+
+    def formula(spm_unit, period, parameters):
+        p = parameters(period).gov.states.hi.dhs.tanf
+
+        unit_size = spm_unit("spm_unit_size", period)
+        capped_size = min_(unit_size, p.max_unit_size)
+
+        # Get Standard of Need for household size
+        son = p.standard_of_need.amount[capped_size]
+
+        # Standard of Assistance = SON × 48%
+        # NOTE: SOA is reduced by 20% after the family receives their initial
+        # two full months of benefits for mandatory work-required households.
+        # This reduction cannot be modeled as PolicyEngine cannot track
+        # cumulative months of benefit receipt.
+        return son * p.standard_of_assistance.rate

policyengine_us/variables/gov/states/hi/dhs/tanf/income/earned/hi_tanf_countable_earned_income.py

@@ -0,0 +1,25 @@
+from policyengine_us.model_api import *
+
+
+class hi_tanf_countable_earned_income(Variable):
+    value_type = float
+    entity = SPMUnit
+    label = "Hawaii TANF countable earned income"
+    unit = USD
+    definition_period = MONTH
+    reference = (
+        "https://humanservices.hawaii.gov/wp-content/uploads/2024/12/Hawaii_TANF_State_Plan_Signed_Certified-Eff_20231001.pdf#page=19",
+    )
+    defined_for = StateCode.HI
+
+    def formula(spm_unit, period, parameters):
+        # Sum person-level countable earned income
+        earned = add(
+            spm_unit, period, ["hi_tanf_countable_earned_income_person"]
+        )
+
+        # Subtract dependent care deduction (work expense)
+        dependent_care = spm_unit("hi_tanf_dependent_care_deduction", period)
+
+        # Floor at 0 - dependent care can't reduce earned below zero
+        return max_(earned - dependent_care, 0)

policyengine_us/variables/gov/states/hi/dhs/tanf/income/earned/hi_tanf_countable_earned_income_person.py

@@ -0,0 +1,30 @@
+from policyengine_us.model_api import *
+
+
+class hi_tanf_countable_earned_income_person(Variable):
+    value_type = float
+    entity = Person
+    label = "Hawaii TANF countable earned income per person"
+    unit = USD
+    definition_period = MONTH
+    reference = (
+        "https://humanservices.hawaii.gov/wp-content/uploads/2024/12/Hawaii_TANF_State_Plan_Signed_Certified-Eff_20231001.pdf#page=19",
+    )
+    defined_for = StateCode.HI
+
+    def formula(person, period, parameters):
+        p = parameters(period).gov.states.hi.dhs.tanf.deductions
+
+        # Start with person's gross earned income (federal baseline)
+        gross_earned = person("tanf_gross_earned_income", period)
+
+        # Step 1: Apply 20% standard deduction
+        after_standard = gross_earned * (1 - p.standard_rate)
+
+        # Step 2: Apply $200 flat rate deduction (per person)
+        after_flat = max_(after_standard - p.flat_amount, 0)
+
+        # Step 3: Apply 55% earned income disregard
+        # NOTE: 55% applies for months 1-24; 36% applies after month 24
+        # 55% is used as default here
+        return after_flat * (1 - p.eidr_rate)

policyengine_us/variables/gov/states/hi/dhs/tanf/income/hi_tanf_countable_income.py

@@ -0,0 +1,15 @@
+from policyengine_us.model_api import *
+
+
+class hi_tanf_countable_income(Variable):
+    value_type = float
+    entity = SPMUnit
+    label = "Hawaii TANF countable income"
+    unit = USD
+    definition_period = MONTH
+    reference = (
+        "https://humanservices.hawaii.gov/wp-content/uploads/2024/12/Hawaii_TANF_State_Plan_Signed_Certified-Eff_20231001.pdf#page=19",
+    )
+    defined_for = StateCode.HI
+
+    adds = ["hi_tanf_countable_earned_income", "tanf_gross_unearned_income"]

policyengine_us/variables/gov/states/hi/dhs/tanf/income/hi_tanf_dependent_care_deduction.py

@@ -0,0 +1,40 @@
+from policyengine_us.model_api import *
+
+
+class hi_tanf_dependent_care_deduction(Variable):
+    value_type = float
+    entity = SPMUnit
+    label = "Hawaii TANF dependent care deduction"
+    unit = USD
+    definition_period = MONTH
+    reference = (
+        "https://humanservices.hawaii.gov/wp-content/uploads/2024/12/Hawaii_TANF_State_Plan_Signed_Certified-Eff_20231001.pdf#page=20",
+    )
+    defined_for = StateCode.HI
+
+    def formula(spm_unit, period, parameters):
+        p = parameters(period).gov.states.hi.dhs.tanf.deductions.dependent_care
+
+        # Get actual childcare expenses (YEAR variable, auto-converts to monthly)
+        expenses = spm_unit("childcare_expenses", period)
+
+        # Count children in the unit for the per-child cap
+        person = spm_unit.members
+        is_child = person("is_child", period.this_year)
+        num_children = spm_unit.sum(is_child)
+
+        # Determine rate based on MAX of adult hours
+        # If either parent works full-time, use full-time rate
+        is_adult = ~person("is_child", period.this_year)
+        hours = person("weekly_hours_worked_before_lsr", period.this_year)
+        # Only consider adult hours (children get 0)
+        adult_hours = where(is_adult, hours, 0)
+        max_adult_hours = spm_unit.max(adult_hours)
+
+        # Full-time = 32+ hours/week
+        # Rate is $175 (full-time) or $165 (part-time) per child
+        rate = p.amount.calc(max_adult_hours)
+        max_deduction = num_children * rate
+
+        # Deduct actual cost up to max per child
+        return min_(expenses, max_deduction)

policyengine_us/variables/gov/states/hi/dhs/tanf/income/hi_tanf_gross_income.py

@@ -0,0 +1,14 @@
+from policyengine_us.model_api import *
+
+
+class hi_tanf_gross_income(Variable):
+    value_type = float
+    entity = SPMUnit
+    label = "Hawaii TANF gross income"
+    unit = USD
+    definition_period = MONTH
+    reference = (
+        "https://humanservices.hawaii.gov/wp-content/uploads/2024/12/Hawaii_TANF_State_Plan_Signed_Certified-Eff_20231001.pdf#page=19",
+    )
+    defined_for = StateCode.HI
+    adds = ["tanf_gross_earned_income", "tanf_gross_unearned_income"]

policyengine_us/variables/gov/states/ia/tax/income/credits/ia_cdcc.py

@@ -18,6 +18,8 @@ class ia_cdcc(Variable):
     def formula(tax_unit, period, parameters):
         # Iowa matches the potential federal credit
         federal_cdcc = tax_unit("cdcc_potential", period)
-        netinc = add(tax_unit, period, ["ia_net_income"])
+        taxable_income = tax_unit("ia_taxable_income", period)
         p = parameters(period).gov.states.ia.tax.income
-        return federal_cdcc * p.credits.child_care.fraction.calc(netinc)
+        return federal_cdcc * p.credits.child_care.fraction.calc(
+            taxable_income
+        )

policyengine_us/variables/gov/states/id/tafi/eligibility/id_tafi_eligible.py

@@ -0,0 +1,37 @@
+from policyengine_us.model_api import *
+
+
+class id_tafi_eligible(Variable):
+    value_type = bool
+    entity = SPMUnit
+    label = "Idaho TAFI eligible"
+    definition_period = MONTH
+    reference = (
+        "https://www.law.cornell.edu/regulations/idaho/IDAPA-16.03.08.100",
+        "https://www.law.cornell.edu/regulations/idaho/IDAPA-16.03.08.254",
+    )
+    defined_for = StateCode.ID
+
+    def formula(spm_unit, period, parameters):
+        p = parameters(period).gov.states.id.tafi
+        # Demographic eligibility - use federal baseline (IDAPA 16.03.08.125)
+        demographic_eligible = spm_unit("is_demographic_tanf_eligible", period)
+
+        # Immigration eligibility - use federal baseline (IDAPA 16.03.08.131)
+        immigration_eligible = (
+            add(spm_unit, period, ["is_citizen_or_legal_immigrant"]) > 0
+        )
+
+        # Resources eligibility (IDAPA 16.03.08.200)
+        resources_eligible = spm_unit("id_tafi_resources_eligible", period)
+
+        # Income eligibility - grant calculation must result in positive amount
+        grant_standard = spm_unit("id_tafi_grant_standard", period)
+        income_eligible = grant_standard > 0
+
+        return (
+            demographic_eligible
+            & immigration_eligible
+            & resources_eligible
+            & income_eligible
+        )

policyengine_us/variables/gov/states/id/tafi/eligibility/id_tafi_resources_eligible.py

@@ -0,0 +1,18 @@
+from policyengine_us.model_api import *
+
+
+class id_tafi_resources_eligible(Variable):
+    value_type = bool
+    entity = SPMUnit
+    label = "Idaho TAFI resources eligible"
+    definition_period = MONTH
+    reference = (
+        "https://www.law.cornell.edu/regulations/idaho/IDAPA-16.03.08.200"
+    )
+    defined_for = StateCode.ID
+
+    def formula(spm_unit, period, parameters):
+        p = parameters(period).gov.states.id.tafi.resources
+        # Per IDAPA 16.03.08.200: Resources must not exceed $5,000
+        resources = spm_unit("spm_unit_assets", period.this_year)
+        return resources <= p.limit

policyengine_us/variables/gov/states/id/tafi/id_tafi.py

@@ -0,0 +1,19 @@
+from policyengine_us.model_api import *
+
+
+class id_tafi(Variable):
+    value_type = float
+    entity = SPMUnit
+    label = "Idaho Temporary Assistance for Families in Idaho (TAFI)"
+    unit = USD
+    definition_period = MONTH
+    reference = (
+        "https://www.law.cornell.edu/regulations/idaho/IDAPA-16.03.08.248"
+    )
+    defined_for = "id_tafi_eligible"
+
+    def formula(spm_unit, period, parameters):
+        p = parameters(period).gov.states.id.tafi
+        grant_standard = spm_unit("id_tafi_grant_standard", period)
+        # Cap at maximum grant
+        return min_(grant_standard, p.maximum_grant)

policyengine_us/variables/gov/states/id/tafi/income/id_tafi_countable_earned_income.py

@@ -0,0 +1,20 @@
+from policyengine_us.model_api import *
+
+
+class id_tafi_countable_earned_income(Variable):
+    value_type = float
+    entity = SPMUnit
+    label = "Idaho TAFI countable earned income"
+    unit = USD
+    definition_period = MONTH
+    reference = (
+        "https://www.law.cornell.edu/regulations/idaho/IDAPA-16.03.08.252"
+    )
+    defined_for = StateCode.ID
+
+    def formula(spm_unit, period, parameters):
+        p = parameters(period).gov.states.id.tafi.income.earned
+        gross_earned = add(spm_unit, period, ["tanf_gross_earned_income"])
+        # Per IDAPA 16.03.08.252: 60% of gross earned income is subtracted
+        # from the work incentive table amount
+        return gross_earned * p.countable_rate

policyengine_us/variables/gov/states/id/tafi/income/id_tafi_countable_unearned_income.py

@@ -0,0 +1,16 @@
+from policyengine_us.model_api import *
+
+
+class id_tafi_countable_unearned_income(Variable):
+    value_type = float
+    entity = SPMUnit
+    label = "Idaho TAFI countable unearned income"
+    unit = USD
+    definition_period = MONTH
+    reference = (
+        "https://www.law.cornell.edu/regulations/idaho/IDAPA-16.03.08.250"
+    )
+    defined_for = StateCode.ID
+
+    # Per IDAPA 16.03.08.250: 100% of unearned income is counted
+    adds = ["tanf_gross_unearned_income"]

policyengine_us/variables/gov/states/id/tafi/income/id_tafi_grant_standard.py

@@ -0,0 +1,31 @@
+from policyengine_us.model_api import *
+
+
+class id_tafi_grant_standard(Variable):
+    value_type = float
+    entity = SPMUnit
+    label = "Idaho TAFI grant standard"
+    unit = USD
+    definition_period = MONTH
+    reference = (
+        "https://www.law.cornell.edu/regulations/idaho/IDAPA-16.03.08.249",
+        "https://www.law.cornell.edu/regulations/idaho/IDAPA-16.03.08.250",
+        "https://www.law.cornell.edu/regulations/idaho/IDAPA-16.03.08.252",
+    )
+    defined_for = StateCode.ID
+
+    def formula(spm_unit, period, parameters):
+        p = parameters(period).gov.states.id.tafi
+        countable_earned = spm_unit("id_tafi_countable_earned_income", period)
+        countable_unearned = spm_unit(
+            "id_tafi_countable_unearned_income", period
+        )
+        work_incentive = spm_unit("id_tafi_work_incentive_amount", period)
+
+        # With earned: work_incentive - earned; Without: maximum_grant
+        base = where(
+            countable_earned > 0,
+            work_incentive - countable_earned,
+            p.maximum_grant,
+        )
+        return max_(base - countable_unearned, 0)

policyengine_us/variables/gov/states/id/tafi/income/id_tafi_work_incentive_amount.py

@@ -0,0 +1,22 @@
+from policyengine_us.model_api import *
+
+
+class id_tafi_work_incentive_amount(Variable):
+    value_type = float
+    entity = SPMUnit
+    label = "Idaho TAFI work incentive table amount"
+    unit = USD
+    definition_period = MONTH
+    reference = (
+        "https://www.law.cornell.edu/regulations/idaho/IDAPA-16.03.08.251"
+    )
+    defined_for = StateCode.ID
+
+    def formula(spm_unit, period, parameters):
+        p = parameters(period).gov.states.id.tafi.work_incentive_table
+        size = spm_unit("spm_unit_size", period.this_year)
+        capped_size = min_(size, p.max_size)
+        base_amount = p.amount[capped_size]
+        additional_people = max_(size - p.max_size, 0)
+        additional_amount = additional_people * p.additional_person
+        return base_amount + additional_amount