Skip to content

Commit 70784e4

Browse files
committed
adding missing fields to flytetask remote entity
1 parent f0ba47f commit 70784e4

File tree

1 file changed

+17
-0
lines changed

1 file changed

+17
-0
lines changed

flytekit/remote/entities.py

+17
Original file line numberDiff line numberDiff line change
@@ -49,7 +49,11 @@ def __init__(
4949
custom,
5050
container=None,
5151
task_type_version: int = 0,
52+
security_context=None,
5253
config=None,
54+
k8s_pod=None,
55+
sql=None,
56+
extended_resources=None,
5357
should_register: bool = False,
5458
):
5559
super(FlyteTask, self).__init__(
@@ -61,7 +65,11 @@ def __init__(
6165
custom,
6266
container=container,
6367
task_type_version=task_type_version,
68+
security_context=security_context,
6469
config=config,
70+
k8s_pod=k8s_pod,
71+
sql=sql,
72+
extended_resources=extended_resources,
6573
)
6674
)
6775
self._should_register = should_register
@@ -146,6 +154,10 @@ def k8s_pod(self):
146154
def sql(self):
147155
return self.template.sql
148156

157+
@property
158+
def extended_resources(self):
159+
return self.template.extended_resources
160+
149161
@property
150162
def should_register(self) -> bool:
151163
return self._should_register
@@ -172,6 +184,11 @@ def promote_from_model(cls, base_model: _task_model.TaskTemplate) -> FlyteTask:
172184
custom=base_model.custom,
173185
container=base_model.container,
174186
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,
175192
)
176193
# Override the newly generated name if one exists in the base model
177194
if not base_model.id.is_empty:

0 commit comments

Comments
 (0)