fix climate zone index

rewiringamerica · Feb 6, 2025 · 3520805 · 3520805
1 parent d65952d
commit 3520805
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/scripts/megastock/write_databricks_to_bigquery_03.py b/scripts/megastock/write_databricks_to_bigquery_03.py
@@ -110,7 +110,7 @@
 # optimize the table by partitioning and clustering
 query = f"""
 CREATE TABLE `{bq_write_path}_optimized`
-PARTITION BY RANGE_BUCKET(climate_zone_int__m, GENERATE_ARRAY(1, {len(climate_zone_mapping)+1}, 1))
+PARTITION BY RANGE_BUCKET(climate_zone_int__m, GENERATE_ARRAY(1, {len(CLIMATE_ZONE_TO_INDEX)+1}, 1))
 CLUSTER BY  heating_fuel__m, geometry_building_type_acs__m, geometry_floor_area__m, vintage__m AS
 SELECT *,
 FROM `{bq_write_path}`