fix: Add additional dependencies for ModelTrainer (5668) #581

Job	Run time
collab-check	6s
wait-for-approval	3s
detect-changes	10s
integ-tests (sagemaker-train)	39m 35s
integ-tests (sagemaker-mlops)	28m 26s
integ-tests (sagemaker-serve)	30m 34s
integ-tests (sagemaker-core)	18m 23s
codestyle-doc-tests (sagemaker-serve)	5m 13s
codestyle-doc-tests (sagemaker-mlops)	5m 28s
codestyle-doc-tests (sagemaker-train)	5m 47s
codestyle-doc-tests (sagemaker-core)	27m 18s
unit-tests (sagemaker-core)	1h 0m 32s
unit-tests (sagemaker-train)	28m 19s
unit-tests (sagemaker-serve)	43m 52s
unit-tests (sagemaker-mlops)	30m 20s
	5h 24m 6s

Provide feedback