style(providers/google): improve BigQueryInsertJobOperator type hinti…

…ng (#36894) * style(providers/google): improve BigQueryInsertJobOperator type hinting * style: resolve mypy failure
apache · Jan 20, 2024 · 09bb1a8 · 09bb1a8
1 parent 6e7085b
commit 09bb1a8
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/airflow/providers/google/cloud/operators/bigquery.py b/airflow/providers/google/cloud/operators/bigquery.py
@@ -2852,6 +2852,7 @@ def execute(self, context: Any):
  persist_kwargs["dataset_id"] = table["datasetId"]
  persist_kwargs["project_id"] = table["projectId"]
  BigQueryTableLink.persist(**persist_kwargs)
+
  self.job_id = job.job_id
 
  if self.project_id:
@@ -2861,6 +2862,7 @@ def execute(self, context: Any):
  location=self.location,
  )
  context["ti"].xcom_push(key="job_id_path", value=job_id_path)
+
  # Wait for the job to complete
  if not self.deferrable:
  job.result(timeout=self.result_timeout, retry=self.result_retry)
@@ -2882,7 +2884,7 @@ def execute(self, context: Any):
  self.log.info("Current state of job %s is %s", job.job_id, job.state)
  self._handle_job_error(job)
 
- def execute_complete(self, context: Context, event: dict[str, Any]):
+ def execute_complete(self, context: Context, event: dict[str, Any]) -> str | None:
  """Callback for when the trigger fires.
 
  This returns immediately. It relies on trigger to throw an exception,