diff --git a/src/helm/clients/bedrock_client.py b/src/helm/clients/bedrock_client.py index e09761dc565..a74c612709c 100644 --- a/src/helm/clients/bedrock_client.py +++ b/src/helm/clients/bedrock_client.py @@ -113,7 +113,7 @@ def __init__( self.tokenizer = tokenizer self.tokenizer_name = tokenizer_name self.bedrock_client = get_bedrock_client_v1( - assumed_role=assumed_role, + assumed_role=assumed_role or os.environ.get("BEDROCK_ASSUME_ROLE", None), region=region, )