Need to select these to get them out.

Author: Piwington

statistical_methods_library/imputation/ratio_calculators.py

@@ -44,9 +44,6 @@ def mean_of_ratios(
         "period",
         "grouping",
         "ref",
-        "link_inclusion_previous",
-        "link_inclusion_current",
-        "link_inclusion_next",
         when(col("link_inclusion_current"), col("aux")).alias("aux"),
         "link_inclusion_previous",
         "link_inclusion_next",
@@ -279,7 +276,8 @@ def upper_bound(c):
         )
 
     growth_df = df.select(
-        "ref", "period", "grouping", *growth_additional_outputs.keys()
+        "ref", "period", "grouping", "link_inclusion_previous", "link_inclusion_next",
+        "link_inclusion_current", *growth_additional_outputs.keys()
     )
 
     return [