Import air quality assemblies, as well as climate assemblies

mysociety · Sep 9, 2024 · 7ec4648 · 7ec4648
1 parent b4194bc
commit 7ec4648
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/caps/management/commands/import_citizens_assemblies.py b/caps/management/commands/import_citizens_assemblies.py
@@ -21,8 +21,11 @@ def get_citizens_assembilies() -> pd.DataFrame:
         done_survey=True,
     )
 
-    # limit to those with 'climate' in the thematic_grouping
-    df = df[df["thematic_grouping"].str.lower().str.contains("climate")]
+    # limit to those with climate-related thematic groups
+    df = df[
+        df["thematic_grouping"].str.contains("climate", case=False)
+        | df["thematic_grouping"].str.contains("air quality", case=False)
+    ]
     # limit to completed assemblies
     df = df[df["assembly_status"] == "Finished"]
     # limit to just those that have a local_authority_code