Skip to content

Commit b97aed9

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

File tree

1 file changed

+8
-5
lines changed

1 file changed

+8
-5
lines changed

src/sagemaker/workflow/airflow.py

Lines changed: 8 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,6 @@ 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:
1179+
return [obj._to_request_dict() for obj in object_list]
1180+
return object_list

0 commit comments

Comments
 (0)