Issue] STS client is not using regional endpoints #32
Description
Our use case is to start the SageMaker training job from the SageMaker Studio notebook and the studio is attached to a private vpc.
When I try to create SSHEstimatorWrapper using below from my studio notebook
ssh_wrapper = SSHEstimatorWrapper.create(pytorch_estimator, connection_wait_time_seconds=0)
I'm getting ConnectTimeoutError: Connect timeout on endpoint URL: "https://sts.amazonaws.com/" exception.
This is because since we have regional vpc endpoint and we can access only regional endpoints like https://sts.us-east-1.amazonaws.com/ . From here I see that this calls only global endpoint
We would need to pass region parameter during the initalization of sts boto3 client so that it uses sts regional endpoints based on teh region