diff --git a/src/sagemaker/session.py b/src/sagemaker/session.py index d704185d87..b84d9f96bf 100644 --- a/src/sagemaker/session.py +++ b/src/sagemaker/session.py @@ -679,7 +679,7 @@ def expand_role(self, role): if '/' in role: return role else: - return boto3.resource("iam").Role(role).arn + return self.boto_session.resource('iam').Role(role).arn def get_caller_identity_arn(self): """Returns the ARN user or role whose credentials are used to call the API.