batch as templated field in DataprocCreateBatchOperator (#20905)

apache · Jan 26, 2022 · 60aa518 · 60aa518
1 parent e45f0e9
commit 60aa518
Showing 1 changed file with 9 additions and 4 deletions.
diff --git a/airflow/providers/google/cloud/operators/dataproc.py b/airflow/providers/google/cloud/operators/dataproc.py
@@ -2032,12 +2032,16 @@ class DataprocCreateBatchOperator(BaseOperator):
  """
  Creates a batch workload.
 
- :param project_id: Required. The ID of the Google Cloud project that the cluster belongs to.
- :param region: Required. The Cloud Dataproc region in which to handle the request.
- :param batch: Required. The batch to create.
+ :param project_id: Required. The ID of the Google Cloud project that the cluster belongs to. (templated)
+ :type project_id: str
+ :param region: Required. The Cloud Dataproc region in which to handle the request. (templated)
+ :type region: str
+ :param batch: Required. The batch to create. (templated)
+ :type batch: google.cloud.dataproc_v1.types.Batch
  :param batch_id: Optional. The ID to use for the batch, which will become the final component
  of the batch's resource name.
- This value must be 4-63 characters. Valid characters are /[a-z][0-9]-/.
+ This value must be 4-63 characters. Valid characters are /[a-z][0-9]-/. (templated)
+ :type batch_id: str
  :param request_id: Optional. A unique id used to identify the request. If the server receives two
  ``CreateBatchRequest`` requests with the same id, then the second request will be ignored and
  the first ``google.longrunning.Operation`` created and stored in the backend is returned.
@@ -2059,6 +2063,7 @@ class DataprocCreateBatchOperator(BaseOperator):
 
  template_fields: Sequence[str] = (
  'project_id',
+ 'batch',
  'batch_id',
  'region',
  'impersonation_chain',