Fix unit and integ tests #39

Job	Run time
collab-check	5s
wait-for-approval	3s
detect-changes	9s
unit-tests (sagemaker-train)	22m 23s
codestyle-doc-tests (sagemaker-core)	7m 33s
codestyle-doc-tests (sagemaker-serve)	6m 11s
unit-tests (sagemaker-serve)	35m 53s
codestyle-doc-tests (sagemaker-train)	5m 57s
unit-tests (sagemaker-mlops)	19m 5s
integ-tests (sagemaker-serve)	28m 1s
codestyle-doc-tests (sagemaker-mlops)	5m 10s
integ-tests (sagemaker-mlops)	19m 17s
integ-tests (sagemaker-train)	50m 26s
unit-tests (sagemaker-core)	16m 40s
integ-tests (sagemaker-core)	10m 13s
	3h 47m 6s

Provide feedback