This is an automated email from the ASF dual-hosted git repository.
shahar1 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 3f7d9c0ae73 Allow templating GCS and BigQuery URIs in Vertex AI batch
prediction (#70679)
3f7d9c0ae73 is described below
commit 3f7d9c0ae730596d1531455fbbf39b6aa5e80cbf
Author: osintalex <[email protected]>
AuthorDate: Sat Sep 12 10:07:16 2026 +0100
Allow templating GCS and BigQuery URIs in Vertex AI batch prediction
(#70679)
---
.../google/cloud/operators/vertex_ai/batch_prediction_job.py | 12 +++++++++++-
1 file changed, 11 insertions(+), 1 deletion(-)
diff --git
a/providers/google/src/airflow/providers/google/cloud/operators/vertex_ai/batch_prediction_job.py
b/providers/google/src/airflow/providers/google/cloud/operators/vertex_ai/batch_prediction_job.py
index 86f15a0c5e5..388485d4efa 100644
---
a/providers/google/src/airflow/providers/google/cloud/operators/vertex_ai/batch_prediction_job.py
+++
b/providers/google/src/airflow/providers/google/cloud/operators/vertex_ai/batch_prediction_job.py
@@ -159,7 +159,17 @@ class
CreateBatchPredictionJobOperator(GoogleCloudBaseOperator):
:param poll_interval: Interval size which defines how often job status is
checked in deferrable mode.
"""
- template_fields = ("region", "project_id", "model_name",
"impersonation_chain", "job_display_name")
+ template_fields = (
+ "region",
+ "project_id",
+ "model_name",
+ "impersonation_chain",
+ "job_display_name",
+ "gcs_source",
+ "bigquery_source",
+ "gcs_destination_prefix",
+ "bigquery_destination_prefix",
+ )
operator_extra_links = (VertexAIBatchPredictionJobLink(),)
def __init__(
