Restore BEDROCK_ASSUME_ROLE

stanford-crfm · Jan 7, 2025 · a8c996f · a8c996f
1 parent 79eda8d
commit a8c996f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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,
         )