Skip to content

Commit ec7175c

Browse files
authored
Update pipeline.py
Fixing bug introduced by a change in the SageMaker SDK. aws/sagemaker-python-sdk#3141
1 parent 9301d2c commit ec7175c

File tree

1 file changed

+2
-6
lines changed

1 file changed

+2
-6
lines changed

ML Pipelines scripts/pipeline.py

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -106,12 +106,8 @@ def get_pipeline(
106106
processing_instance_count = ParameterInteger(
107107
name="ProcessingInstanceCount", default_value=1
108108
)
109-
processing_instance_type = ParameterString(
110-
name="ProcessingInstanceType", default_value="ml.m5.xlarge"
111-
)
112-
training_instance_type = ParameterString(
113-
name="TrainingInstanceType", default_value="ml.m5.xlarge"
114-
)
109+
processing_instance_type = "ml.m5.xlarge"
110+
training_instance_type = "ml.m5.xlarge"
115111
model_approval_status = ParameterString(
116112
name="ModelApprovalStatus",
117113
default_value="PendingManualApproval", # ModelApprovalStatus can be set to a default of "Approved" if you don't want manual approval.

0 commit comments

Comments
 (0)