From 285a3733a968bbf0c94e76d0ff5d61a80e7429af Mon Sep 17 00:00:00 2001
From: Krrish Dholakia <krrishdholakia@gmail.com>
Date: Tue, 30 Apr 2024 12:14:29 -0700
Subject: [PATCH] test(test_image_generation.py): fix test

---
 litellm/tests/test_image_generation.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/litellm/tests/test_image_generation.py b/litellm/tests/test_image_generation.py
index 31e7d3d90..82068a115 100644
--- a/litellm/tests/test_image_generation.py
+++ b/litellm/tests/test_image_generation.py
@@ -137,7 +137,7 @@ def test_image_generation_bedrock():
         response = litellm.image_generation(
             prompt="A cute baby sea otter",
             model="bedrock/stability.stable-diffusion-xl-v1",
-            aws_region_name="us-east-1",
+            aws_region_name="us-west-2",
         )
         print(f"response: {response}")
     except litellm.RateLimitError as e:
@@ -157,7 +157,7 @@ async def test_aimage_generation_bedrock_with_optional_params():
         response = await litellm.aimage_generation(
             prompt="A cute baby sea otter",
             model="bedrock/stability.stable-diffusion-xl-v1",
-            size="128x128",
+            size="256x256",
         )
         print(f"response: {response}")
     except litellm.RateLimitError as e: