Fix unit and integ tests #40

Job	Run time
collab-check	3s
wait-for-approval	4s
detect-changes	11s
codestyle-doc-tests (sagemaker-train)	6m 0s
codestyle-doc-tests (sagemaker-serve)	6m 15s
codestyle-doc-tests (sagemaker-mlops)	5m 10s
unit-tests (sagemaker-serve)	35m 29s
codestyle-doc-tests (sagemaker-core)	27m 25s
integ-tests (sagemaker-core)	12m 12s
integ-tests (sagemaker-serve)	28m 22s
integ-tests (sagemaker-train)	49m 58s
unit-tests (sagemaker-train)	21m 50s
unit-tests (sagemaker-mlops)	25m 17s
integ-tests (sagemaker-mlops)	20m 48s
unit-tests (sagemaker-core)	17m 15s
	4h 16m 19s

Provide feedback