Update dataproc.py

Adding DataprocBatchLink.persist to display link for createbatch operator
apache · Jul 8, 2024 · c7f1db7 · c7f1db7
1 parent be27182
commit c7f1db7
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/airflow/providers/google/cloud/operators/dataproc.py b/airflow/providers/google/cloud/operators/dataproc.py
@@ -3055,6 +3055,13 @@ def execute(self, context: Context):
  self.log.info("Batch %s created", self.batch_id)
 
  else:
+ DataprocBatchLink.persist(
+ context=context,
+ operator=self,
+ project_id=self.project_id,
+ region=self.region,
+ batch_id=self.batch_id,
+ )
  return self.operation.operation.name
 
  else: