Fix: check for 'area' column

mysociety · Nov 15, 2023 · ae67f86 · ae67f86
1 parent e2a12cc
commit ae67f86
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/hub/management/commands/generate_wi_groups_csv.py b/hub/management/commands/generate_wi_groups_csv.py
@@ -100,11 +100,10 @@ def process_data(self, df):
 
         if not self._quiet:
             self.stdout.write("Generating GSS codes for new WI groups")
-
         df["area"] = df.progress_apply(
             lambda row: self.process_lat_long(row.lat_lon, row.group_name)
-            if pd.isna(row.area)
-            else row.area,
+            if "area" not in row
+            else row["area"],
             axis=1,
         )
         return df