@@ -166,6 +166,7 @@ def test_execution_spec(literal_value_pair):
166
166
),
167
167
raw_output_data_config = _common_models .RawOutputDataConfig (output_location_prefix = "raw_output" ),
168
168
max_parallelism = 100 ,
169
+ interruptible = True
169
170
)
170
171
assert obj .launch_plan .resource_type == _identifier .ResourceType .LAUNCH_PLAN
171
172
assert obj .launch_plan .domain == "domain"
@@ -183,6 +184,7 @@ def test_execution_spec(literal_value_pair):
183
184
]
184
185
assert obj .disable_all is None
185
186
assert obj .max_parallelism == 100
187
+ assert obj .interruptible == True
186
188
assert obj .raw_output_data_config .output_location_prefix == "raw_output"
187
189
188
190
obj2 = _execution .ExecutionSpec .from_flyte_idl (obj .to_flyte_idl ())
@@ -203,6 +205,7 @@ def test_execution_spec(literal_value_pair):
203
205
]
204
206
assert obj2 .disable_all is None
205
207
assert obj2 .max_parallelism == 100
208
+ assert obj2 .interruptible == True
206
209
assert obj2 .raw_output_data_config .output_location_prefix == "raw_output"
207
210
208
211
obj = _execution .ExecutionSpec (
@@ -220,6 +223,7 @@ def test_execution_spec(literal_value_pair):
220
223
assert obj .metadata .principal == "tester"
221
224
assert obj .notifications is None
222
225
assert obj .disable_all is True
226
+ assert obj .interruptible is None
223
227
224
228
obj2 = _execution .ExecutionSpec .from_flyte_idl (obj .to_flyte_idl ())
225
229
assert obj == obj2
@@ -233,6 +237,7 @@ def test_execution_spec(literal_value_pair):
233
237
assert obj2 .metadata .principal == "tester"
234
238
assert obj2 .notifications is None
235
239
assert obj2 .disable_all is True
240
+ assert obj2 .interruptible is None
236
241
237
242
238
243
def test_workflow_execution_data_response ():
0 commit comments