Skip to content

Commit 40d6656

Browse files
committed
fix: Nullable fields in processing_config
1 parent 1872483 commit 40d6656

File tree

1 file changed

+10
-5
lines changed

1 file changed

+10
-5
lines changed

src/sagemaker/workflow/airflow.py

Lines changed: 10 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1154,10 +1154,11 @@ def processing_config(
11541154
)
11551155
config["ProcessingResources"] = processing_resources
11561156

1157-
stopping_condition = sagemaker.processing.ProcessingJob.prepare_stopping_condition(
1158-
processor.max_runtime_in_seconds
1159-
)
1160-
config["StoppingCondition"] = stopping_condition
1157+
if processor.max_runtime_in_seconds is not None:
1158+
stopping_condition = sagemaker.processing.ProcessingJob.prepare_stopping_condition(
1159+
processor.max_runtime_in_seconds
1160+
)
1161+
config["StoppingCondition"] = stopping_condition
11611162

11621163
if processor.tags is not None:
11631164
config["Tags"] = processor.tags
@@ -1174,4 +1175,8 @@ def input_output_list_converter(object_list):
11741175
Returns:
11751176
List of dicts
11761177
"""
1177-
return [obj._to_request_dict() for obj in object_list]
1178+
if object_list is not None:
1179+
dict_list = [obj._to_request_dict() for obj in object_list]
1180+
else:
1181+
dict_list = object_list
1182+
return dict_list

0 commit comments

Comments
 (0)