Change _lsr to _behavioral_response

Fixes PolicyEngine#5258
Mshuning · Oct 31, 2024 · 86189dc · 86189dc
1 parent 7203d33
commit 86189dc
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/policyengine_us/variables/gov/simulation/labor_supply_response/labor_supply_response.py b/policyengine_us/variables/gov/simulation/labor_supply_response/labor_supply_response.py
@@ -11,7 +11,9 @@ class relative_income_change(Variable):
 
     def formula(person, period, parameters):
         simulation = person.simulation
-        measurement_branch = simulation.get_branch("lsr_measurement")
+        measurement_branch = simulation.get_branch(
+            "behavioral_response_measurement"
+        )
         baseline_branch = simulation.get_branch("baseline").get_branch(
             "baseline_behavioral_response_measurement"
         )
@@ -47,7 +49,9 @@ class relative_wage_change(Variable):
 
     def formula(person, period, parameters):
         simulation = person.simulation
-        measurement_branch = simulation.get_branch("lsr_measurement")
+        measurement_branch = simulation.get_branch(
+            "behavioral_response_measurement"
+        )
         baseline_branch = simulation.get_branch("baseline").get_branch(
             "baseline_behavioral_response_measurement"
         )
@@ -224,7 +228,7 @@ def formula(person, period, parameters):
             return 0
 
         measurement_branch = simulation.get_branch(
-            "lsr_measurement", clone_system=True
+            "behavioral_response_measurement", clone_system=True
         )  # A branch without LSRs
         baseline_branch = simulation.get_branch("baseline").get_branch(
             "baseline_behavioral_response_measurement", clone_system=True