QueryJobConfig.labels instead of job_labels

allegro · May 24, 2024 · e63472f · e63472f
1 parent 9143fa0
commit e63472f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/bigflow/bigquery/dataset_manager.py b/bigflow/bigquery/dataset_manager.py
@@ -288,7 +288,7 @@ def write(self, table_id: str, sql: str, mode: str) -> 'google.cloud.bigquery.ta
         job_config.write_disposition = mode
 
         if self.job_labels:
-            job_config.job_labels = self.job_labels
+            job_config.labels = self.job_labels
 
         job = self.bigquery_client.query(sql, job_config=job_config)
         return job.result()
@@ -323,7 +323,7 @@ def collect(self, sql: str) -> 'pandas.DataFrame':
         job_config = bigquery.QueryJobConfig()
 
         if self.job_labels:
-            job_config.job_labels = self.job_labels
+            job_config.labels = self.job_labels
 
         return self._query(sql, job_config=job_config).to_dataframe()