fix

eqasim-org · Feb 5, 2024 · 6ca0be1 · 6ca0be1
1 parent 17ecf7c
commit 6ca0be1
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/data/census/filtered.py b/data/census/filtered.py
@@ -15,6 +15,7 @@ def execute(context):
     df = context.stage("data.census.cleaned")
 
     # Household size
+    df_codes = context.stage("data.spatial.codes")
     df_size = df[["household_id"]].groupby("household_id").size().reset_index(name = "household_size2")
     df = pd.merge(df_codes, df_size)