Skip to content

Commit 9bcdada

Browse files
authored
Add 'main' param to template_fields in DataprocSubmitPySparkJobOperator (#9154)
1 parent 32ef0cd commit 9bcdada

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

airflow/providers/google/cloud/operators/dataproc.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1242,7 +1242,7 @@ class DataprocSubmitPySparkJobOperator(DataprocJobBaseOperator):
12421242
Start a PySpark Job on a Cloud DataProc cluster.
12431243
12441244
:param main: [Required] The Hadoop Compatible Filesystem (HCFS) URI of the main
1245-
Python file to use as the driver. Must be a .py file.
1245+
Python file to use as the driver. Must be a .py file. (templated)
12461246
:type main: str
12471247
:param arguments: Arguments for the job. (templated)
12481248
:type arguments: list
@@ -1256,7 +1256,7 @@ class DataprocSubmitPySparkJobOperator(DataprocJobBaseOperator):
12561256
:type pyfiles: list
12571257
"""
12581258

1259-
template_fields = ['arguments', 'job_name', 'cluster_name',
1259+
template_fields = ['main', 'arguments', 'job_name', 'cluster_name',
12601260
'region', 'dataproc_jars', 'dataproc_properties']
12611261
ui_color = '#0273d4'
12621262
job_type = 'pyspark_job'

0 commit comments

Comments
 (0)