Fix minor bug

PolicyEngine · Nov 30, 2024 · 6bd1e49 · 6bd1e49
1 parent 381150b
commit 6bd1e49
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/policyengine_uk_data/datasets/frs/local_areas/constituencies/calibrate.py b/policyengine_uk_data/datasets/frs/local_areas/constituencies/calibrate.py
@@ -84,7 +84,7 @@ def loss(w):
 def update_weights(weights, mapping_matrix):
     mapping_matrix = mapping_matrix.set_index(mapping_matrix.columns[0])
     mapping_matrix = mapping_matrix.div(mapping_matrix.sum(), axis=1)
-    mapped_weights = mapping_matrix.T.dot(weights)
+    mapped_weights = mapping_matrix.T.dot(weights).values
     return mapped_weights[mapping_matrix.columns.argsort(), :]