[AIRFLOW-2016] assign template_fields for Dataproc Workflow Template

sub-classes, not base class

Closes #3346 from mchalek/mchalek-fix-inline-
workflow-template-operator
This commit is contained in:
Kevin McHale 2018-05-10 22:33:19 +01:00 коммит произвёл Kaxil Naik
Родитель 7ca45b252b
Коммит 961aa05812
1 изменённых файлов: 4 добавлений и 2 удалений

Просмотреть файл

@ -1031,8 +1031,6 @@ class DataProcPySparkOperator(BaseOperator):
class DataprocWorkflowTemplateBaseOperator(BaseOperator):
template_fields = ['template_id', 'template']
@apply_defaults
def __init__(self,
project_id,
@ -1083,6 +1081,8 @@ class DataprocWorkflowTemplateInstantiateOperator(DataprocWorkflowTemplateBaseOp
:type delegate_to: string
"""
template_fields = ['template_id']
@apply_defaults
def __init__(self, template_id, *args, **kwargs):
(super(DataprocWorkflowTemplateInstantiateOperator, self)
@ -1125,6 +1125,8 @@ class DataprocWorkflowTemplateInstantiateInlineOperator(
:type delegate_to: string
"""
template_fields = ['template']
@apply_defaults
def __init__(self, template, *args, **kwargs):
(super(DataprocWorkflowTemplateInstantiateInlineOperator, self)