Add fix to MTR bug

jdebacker · Aug 9, 2021 · e2923c5 · e2923c5
1 parent b570997
commit e2923c5
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/openfisca_uk/graphs/hypothetical.py b/openfisca_uk/graphs/hypothetical.py
@@ -102,12 +102,20 @@ def mtr_chart(
         reforms = [reforms]
         names = ["Reform"]
 
+    test_sim = IndividualSim()
+    situation_function(test_sim)
+    situation_data = test_sim.situation_data
+    households = situation_data["households"]
+    adults = households[list(households.keys())[0]]["adults"]
+    primary_adult = adults[0]
+
     invert += INVERTED_DERIV_VARIABLES
     df = get_wide_reform_individual_data(
         reforms,
         names,
         variables,
         situation_function,
+        primary_adult_name=primary_adult,
         include_derivatives=True,
     )
     for var in df.columns: