Skip to content

Commit 0ba9498

Browse files
committed
fix: formatting in bedrock test
1 parent 6edab4c commit 0ba9498

File tree

1 file changed

+9
-7
lines changed

1 file changed

+9
-7
lines changed

adalflow/tests/test_aws_bedrock_client.py

Lines changed: 9 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,6 @@
44

55
from adalflow.core.types import ModelType, GeneratorOutput
66
from adalflow.components.model_client import BedrockAPIClient
7-
from adalflow import setup_env
87

98

109
def getenv_side_effect(key):
@@ -21,12 +20,15 @@ def getenv_side_effect(key):
2120
class TestBedrockClient(unittest.TestCase):
2221
def setUp(self):
2322
# Patch os.environ to ensure all environment variables are set
24-
with patch.dict(os.environ, {
25-
"AWS_ACCESS_KEY_ID": "fake_api_key",
26-
"AWS_SECRET_ACCESS_KEY": "fake_api_key",
27-
"AWS_REGION_NAME": "fake_api_key",
28-
"AWS_PROFILE_NAME": "fake_profile" # Adding additional profile if needed
29-
}):
23+
with patch.dict(
24+
os.environ,
25+
{
26+
"AWS_ACCESS_KEY_ID": "fake_api_key",
27+
"AWS_SECRET_ACCESS_KEY": "fake_api_key",
28+
"AWS_REGION_NAME": "fake_api_key",
29+
"AWS_PROFILE_NAME": "fake_profile", # Adding additional profile if needed
30+
},
31+
):
3032
# Now patch os.getenv to return mocked environment variable values
3133
with patch("os.getenv", side_effect=getenv_side_effect):
3234
self.client = BedrockAPIClient()

0 commit comments

Comments
 (0)