From e63472f9f5adc0e7946e6b67949f0b68f1a04b23 Mon Sep 17 00:00:00 2001 From: Artur Nowosielski Date: Tue, 23 Apr 2024 10:00:33 +0200 Subject: [PATCH] QueryJobConfig.labels instead of job_labels --- bigflow/bigquery/dataset_manager.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/bigflow/bigquery/dataset_manager.py b/bigflow/bigquery/dataset_manager.py index 4c54b9e3..51252bbc 100644 --- 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()