Fix unit and integ tests #36

Job	Run time
collab-check	3s
wait-for-approval	3s
detect-changes	12s
codestyle-doc-tests (sagemaker-core)	20m 2s
codestyle-doc-tests (sagemaker-train)	6m 58s
unit-tests (sagemaker-serve)	27m 34s
unit-tests (sagemaker-train)	20m 50s
codestyle-doc-tests (sagemaker-mlops)	7m 41s
unit-tests (sagemaker-core)	22m 39s
integ-tests (sagemaker-serve)	28m 56s
codestyle-doc-tests (sagemaker-serve)	7m 29s
integ-tests (sagemaker-mlops)	20m 9s
integ-tests (sagemaker-core)	11m 13s
integ-tests (sagemaker-train)	50m 25s
unit-tests (sagemaker-mlops)	23m 55s
	4h 8m 9s

Provide feedback