This is an automated email from the ASF dual-hosted git repository.
shahar 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 4a61eebddb5 fix poll_sleep is ignored in
DataflowStartFlexTemplateOperator definition (#58846)
4a61eebddb5 is described below
commit 4a61eebddb584d3dbbf864f3503e72a1e04f6223
Author: Henry Chen <[email protected]>
AuthorDate: Sat Nov 29 22:19:26 2025 +0800
fix poll_sleep is ignored in DataflowStartFlexTemplateOperator definition
(#58846)
---
.../google/src/airflow/providers/google/cloud/operators/dataflow.py | 1 +
providers/google/tests/unit/google/cloud/operators/test_dataflow.py | 1 +
2 files changed, 2 insertions(+)
diff --git
a/providers/google/src/airflow/providers/google/cloud/operators/dataflow.py
b/providers/google/src/airflow/providers/google/cloud/operators/dataflow.py
index 467ef696401..3db9c6417d4 100644
--- a/providers/google/src/airflow/providers/google/cloud/operators/dataflow.py
+++ b/providers/google/src/airflow/providers/google/cloud/operators/dataflow.py
@@ -583,6 +583,7 @@ class
DataflowStartFlexTemplateOperator(GoogleCloudBaseOperator):
def hook(self) -> DataflowHook:
hook = DataflowHook(
gcp_conn_id=self.gcp_conn_id,
+ poll_sleep=self.poll_sleep,
drain_pipeline=self.drain_pipeline,
cancel_timeout=self.cancel_timeout,
wait_until_finished=self.wait_until_finished,
diff --git
a/providers/google/tests/unit/google/cloud/operators/test_dataflow.py
b/providers/google/tests/unit/google/cloud/operators/test_dataflow.py
index 58d188780df..acd766b7b52 100644
--- a/providers/google/tests/unit/google/cloud/operators/test_dataflow.py
+++ b/providers/google/tests/unit/google/cloud/operators/test_dataflow.py
@@ -305,6 +305,7 @@ class TestDataflowStartFlexTemplateOperator:
cancel_timeout=600,
wait_until_finished=None,
impersonation_chain=None,
+ poll_sleep=10,
)
mock_dataflow.return_value.start_flex_template.assert_called_once_with(
body={"launchParameter": TEST_FLEX_PARAMETERS},
