feat(train): Add wait_timeout parameter to train() #697

Job	Run time
collab-check	4s
wait-for-approval	3s
detect-changes	8s
integ-tests (sagemaker-train)	39m 50s
integ-tests (sagemaker-mlops)	33m 30s
codestyle-doc-tests (sagemaker-train)	6m 45s
codestyle-doc-tests (sagemaker-serve)	6m 59s
unit-tests (sagemaker-mlops)	37m 38s
integ-tests (sagemaker-serve)	30m 19s
unit-tests (sagemaker-serve)	49m 27s
unit-tests (sagemaker-train)	25m 56s
codestyle-doc-tests (sagemaker-mlops)	7m 59s
	3h 58m 38s

Provide feedback