fix: Model builder unable to (5667) #571

Job	Run time
collab-check	4s
wait-for-approval	3s
detect-changes	11s
codestyle-doc-tests (sagemaker-mlops)	5m 26s
unit-tests (sagemaker-serve)	43m 27s
codestyle-doc-tests (sagemaker-serve)	6m 0s
unit-tests (sagemaker-mlops)	31m 34s
integ-tests (sagemaker-mlops)	28m 47s
integ-tests (sagemaker-serve)	27m 52s
	2h 23m 24s

Provide feedback