@@ -49,7 +49,11 @@ def __init__(
49
49
custom ,
50
50
container = None ,
51
51
task_type_version : int = 0 ,
52
+ security_context = None ,
52
53
config = None ,
54
+ k8s_pod = None ,
55
+ sql = None ,
56
+ extended_resources = None ,
53
57
should_register : bool = False ,
54
58
):
55
59
super (FlyteTask , self ).__init__ (
@@ -61,7 +65,11 @@ def __init__(
61
65
custom ,
62
66
container = container ,
63
67
task_type_version = task_type_version ,
68
+ security_context = security_context ,
64
69
config = config ,
70
+ k8s_pod = k8s_pod ,
71
+ sql = sql ,
72
+ extended_resources = extended_resources ,
65
73
)
66
74
)
67
75
self ._should_register = should_register
@@ -146,6 +154,10 @@ def k8s_pod(self):
146
154
def sql (self ):
147
155
return self .template .sql
148
156
157
+ @property
158
+ def extended_resources (self ):
159
+ return self .template .extended_resources
160
+
149
161
@property
150
162
def should_register (self ) -> bool :
151
163
return self ._should_register
@@ -172,6 +184,11 @@ def promote_from_model(cls, base_model: _task_model.TaskTemplate) -> FlyteTask:
172
184
custom = base_model .custom ,
173
185
container = base_model .container ,
174
186
task_type_version = base_model .task_type_version ,
187
+ security_context = base_model .security_context ,
188
+ config = base_model .config ,
189
+ k8s_pod = base_model .k8s_pod ,
190
+ sql = base_model .sql ,
191
+ extended_resources = base_model .extended_resources ,
175
192
)
176
193
# Override the newly generated name if one exists in the base model
177
194
if not base_model .id .is_empty :
0 commit comments