feat: added queries on huge countries in talk to ipcc

climateqa/engine/talk_to_data/ipcc/config.py

@@ -51,7 +51,9 @@ MACRO_COUNTRIES = ['JP',
  'ZA',
  'NZ',
  'TF',
- 'CL',
+]
+
+HUGE_MACRO_COUNTRIES = ['CL',
  'CA',
  'AU',
  'US',

climateqa/engine/talk_to_data/ipcc/plots.py

@@ -154,7 +154,7 @@ def plot_choropleth_map_of_country_indicator_for_specific_year(
 
         fig.update_layout(
             mapbox_style="open-street-map",
-            mapbox_zoom=3,
+            mapbox_zoom=2,
             height=800,
             mapbox_center={
                 "lat": latitudes[len(latitudes)//2] if latitudes else 0,

climateqa/engine/talk_to_data/ipcc/queries.py

@@ -1,6 +1,6 @@
 from typing import TypedDict, Optional
 
-from climateqa.engine.talk_to_data.ipcc.config import IPCC_DATASET_URL, MACRO_COUNTRIES
+from climateqa.engine.talk_to_data.ipcc.config import HUGE_MACRO_COUNTRIES, IPCC_DATASET_URL, MACRO_COUNTRIES
 
 class IndicatorPerYearAtLocationQueryParams(TypedDict, total=False):
     """
@@ -50,6 +50,14 @@ def indicator_per_year_at_location_query(
         GROUP BY scenario, year
         ORDER BY year, scenario
         """
+    elif country_code in HUGE_MACRO_COUNTRIES:
+        table_path = f"'{IPCC_DATASET_URL}/{table.lower()}/{country_code}_macro.parquet'"
+        sql_query = f"""
+        SELECT year, scenario, {indicator_column},
+        FROM {table_path}
+        WHERE latitude = {latitude} AND longitude = {longitude} AND year >= 1950
+        ORDER year, scenario
+        """
     else:
         table_path = f"'{IPCC_DATASET_URL}/{table.lower()}/{country_code}.parquet'"
         sql_query = f"""
@@ -109,6 +117,14 @@ def indicator_for_given_year_query(
         GROUP BY latitude, longitude, scenario
         ORDER BY latitude, longitude, scenario
         """
+    elif country_code in HUGE_MACRO_COUNTRIES:
+        table_path = f"'{IPCC_DATASET_URL}/{table.lower()}/{country_code}_macro.parquet'"
+        sql_query = f"""
+        SELECT latitude, longitude, scenario, {indicator_column},
+        FROM {table_path}
+        WHERE year = {year}
+        ORDER BY latitude, longitude, scenario
+        """
     else:
         table_path = f"'{IPCC_DATASET_URL}/{table.lower()}/{country_code}.parquet'"
         sql_query = f"""