QueryJobConfig.labels instead of job_labels

Author: artnowo-alle

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()