diff --git a/tests/llm_translation/test_bedrock_embedding.py b/tests/llm_translation/test_bedrock_embedding.py index 4adb38e867..15a615b8cc 100644 --- a/tests/llm_translation/test_bedrock_embedding.py +++ b/tests/llm_translation/test_bedrock_embedding.py @@ -85,13 +85,13 @@ def test_e2e_bedrock_embedding(): """ print("Testing text embedding...") original_region_name = os.environ.get("AWS_REGION_NAME") - os.environ.pop("AWS_REGION_NAME") + + os.environ["AWS_REGION_NAME"] = "us-east-1" litellm._turn_on_debug() response = litellm.embedding( model="bedrock/us.twelvelabs.marengo-embed-2-7-v1:0", input=["Hello world from LiteLLM with TwelveLabs Marengo!"], - aws_region_name="us-east-1" ) # Validate response structure @@ -128,7 +128,7 @@ def test_e2e_bedrock_embedding_image_twelvelabs_marengo(): """ print("Testing image embedding...") original_region_name = os.environ.get("AWS_REGION_NAME") - os.environ.pop("AWS_REGION_NAME") + os.environ["AWS_REGION_NAME"] = "us-east-1" litellm._turn_on_debug() # Load duck.png and convert to base64