Fixing derive_estimation_variables test

ONSdigital · Jan 28, 2025 · 6234626 · 6234626
1 parent ada4855
commit 6234626
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mbs_results/estimation/pre_processing_estimation.py b/mbs_results/estimation/pre_processing_estimation.py
@@ -100,7 +100,7 @@ def derive_estimation_variables(
 
     """
 
-    population_frame = convert_cell_number(population_frame, cell_number)
+    population_frame = convert_cell_number(population_frame, cell_number).drop(columns = "ni_gb_cell_number")
 
     population_frame = population_frame.merge(
         calibration_group_map, on=[cell_number], how="left"