From d25b079caf3c7ecb11d7c1ea498bd3368a9906c2 Mon Sep 17 00:00:00 2001
From: Ishaan Jaff <ishaanjaffer0324@gmail.com>
Date: Sat, 29 Jun 2024 20:54:22 -0700
Subject: [PATCH] fix img gen test

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

diff --git a/litellm/tests/test_router.py b/litellm/tests/test_router.py
index 5f22e88a17..7c59611d73 100644
--- a/litellm/tests/test_router.py
+++ b/litellm/tests/test_router.py
@@ -1081,7 +1081,7 @@ async def test_aimg_gen_on_router():
             {
                 "model_name": "dall-e-3",
                 "litellm_params": {
-                    "model": "openai/dall-e-3",
+                    "model": "dall-e-3",
                 },
             },
             {
@@ -1137,7 +1137,7 @@ def test_img_gen_on_router():
     try:
         model_list = [
             {
-                "model_name": "openai/dall-e-3",
+                "model_name": "dall-e-3",
                 "litellm_params": {
                     "model": "dall-e-3",
                 },