From a8c996f8b7d81bb3e50451344db50b5ff171f502 Mon Sep 17 00:00:00 2001 From: Yifan Mai Date: Tue, 7 Jan 2025 15:26:19 -0800 Subject: [PATCH] Restore BEDROCK_ASSUME_ROLE --- src/helm/clients/bedrock_client.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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, )