This is an automated email from the ASF dual-hosted git repository.
potiuk pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/airflow.git
The following commit(s) were added to refs/heads/main by this push:
new f12714d0c7 Update dataproc.py (#40643)
f12714d0c7 is described below
commit f12714d0c7d0f6a8a5d7b6b6910c23dce4d8c71a
Author: kandharvishnu <[email protected]>
AuthorDate: Tue Jul 9 01:32:55 2024 +0530
Update dataproc.py (#40643)
Adding DataprocBatchLink.persist to display link for createbatch operator
---
airflow/providers/google/cloud/operators/dataproc.py | 7 +++++++
1 file changed, 7 insertions(+)
diff --git a/airflow/providers/google/cloud/operators/dataproc.py
b/airflow/providers/google/cloud/operators/dataproc.py
index 700f69494b..889f8b3e59 100644
--- a/airflow/providers/google/cloud/operators/dataproc.py
+++ b/airflow/providers/google/cloud/operators/dataproc.py
@@ -3055,6 +3055,13 @@ class
DataprocCreateBatchOperator(GoogleCloudBaseOperator):
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:
