Fix unit and integ tests #46

Job	Run time
collab-check	5s
wait-for-approval	3s
detect-changes	10s
unit-tests (sagemaker-serve)	39m 44s
unit-tests (sagemaker-train)	23m 23s
unit-tests (sagemaker-core)	27m 52s
unit-tests (sagemaker-mlops)	23m 54s
codestyle-doc-tests (sagemaker-mlops)	5m 10s
codestyle-doc-tests (sagemaker-train)	5m 58s
codestyle-doc-tests (sagemaker-core)	19m 22s
codestyle-doc-tests (sagemaker-serve)	6m 15s
integ-tests (sagemaker-serve)	29m 22s
integ-tests (sagemaker-train)	50m 35s
integ-tests (sagemaker-core)	12m 7s
integ-tests (sagemaker-mlops)	20m 51s
	4h 24m 51s

Provide feedback