Merge pull request #95 from CartoDB/jarroyo/cluster-quadbin-tables

Cluster "quadbin raster" table by quadbin
CartoDB · Jan 26, 2023 · 19c8a19 · 19c8a19
2 parents 6c91bb1 + 97481e7
commit 19c8a19
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/raster_loader/io.py b/raster_loader/io.py
@@ -320,8 +320,14 @@ def records_to_bigquery(
 
     data_df = pd.DataFrame(records)
 
+    job_config = bigquery.LoadJobConfig()
+
+    if "quadbin" in data_df.keys():
+        # Cluster table by quadbin
+        job_config.clustering_fields = ["quadbin"]
+
     return client.load_table_from_dataframe(
-        data_df, f"{project_id}.{dataset_id}.{table_id}"
+        data_df, f"{project_id}.{dataset_id}.{table_id}", job_config=job_config
     )