feat: updated sql query for not macro countries in talk to ipcc

climateqa/engine/talk_to_data/ipcc/queries.py

@@ -52,9 +52,14 @@ def indicator_per_year_at_location_query(
     else:
         table_path = f"'{IPCC_DATASET_URL}/{table.lower()}/{country_code}.parquet'"
         sql_query = f"""
-        SELECT year, scenario, AVG({indicator_column}) AS {indicator_column}
-        FROM {table_path}
-        WHERE latitude = {latitude} AND longitude = {longitude} AND year >= 1950
+        WITH medians_per_month AS (
+            SELECT year, scenario, month, MEDIAN({indicator_column}) AS median_value
+            FROM {table_path}
+            WHERE latitude = {latitude} AND longitude = {longitude} AND year >= 1950
+            GROUP BY scenario, year, month
+        )
+        SELECT year, scenario, AVG(median_value) AS {indicator_column}
+        FROM medians_per_month
         GROUP BY scenario, year
         ORDER BY year, scenario
         """
@@ -107,10 +112,16 @@ def indicator_for_given_year_query(
     else:
         table_path = f"'{IPCC_DATASET_URL}/{table.lower()}/{country_code}.parquet'"
         sql_query = f"""
-        SELECT AVG({indicator_column}) AS {indicator_column}, latitude, longitude, scenario
-        FROM {table_path}
-        WHERE year = {year}
+        WITH medians_per_month AS (
+            SELECT latitude, longitude, scenario, month, MEDIAN({indicator_column}) AS median_value
+            FROM {table_path}
+            WHERE year = {year}
+            GROUP BY latitude, longitude, scenario, month
+        )
+        SELECT latitude, longitude, scenario, AVG(median_value) AS {indicator_column}
+        FROM medians_per_month
         GROUP BY latitude, longitude, scenario
         ORDER BY latitude, longitude, scenario
         """
+
     return sql_query.strip()