From 4de19cb116fe626b6f0e23ea345978987e4d88ab Mon Sep 17 00:00:00 2001
From: Krrish Dholakia <krrishdholakia@gmail.com>
Date: Fri, 19 Jul 2024 19:00:55 -0700
Subject: [PATCH] test(test_router_timeout.py): fix test

---
 litellm/tests/test_router_timeout.py | 16 ++++++++++++----
 1 file changed, 12 insertions(+), 4 deletions(-)

diff --git a/litellm/tests/test_router_timeout.py b/litellm/tests/test_router_timeout.py
index 1126f6fb8..ccba7f676 100644
--- a/litellm/tests/test_router_timeout.py
+++ b/litellm/tests/test_router_timeout.py
@@ -1,8 +1,12 @@
 #### What this tests ####
 # This tests if the router timeout error handling during fallbacks
 
-import sys, os, time
-import traceback, asyncio
+import asyncio
+import os
+import sys
+import time
+import traceback
+
 import pytest
 
 sys.path.insert(
@@ -12,9 +16,10 @@ sys.path.insert(
 
 import os
 
+from dotenv import load_dotenv
+
 import litellm
 from litellm import Router
-from dotenv import load_dotenv
 
 load_dotenv()
 
@@ -37,6 +42,7 @@ def test_router_timeouts():
             "litellm_params": {
                 "model": "claude-instant-1.2",
                 "api_key": "os.environ/ANTHROPIC_API_KEY",
+                "mock_response": "hello world",
             },
             "tpm": 20000,
         },
@@ -90,7 +96,9 @@ def test_router_timeouts():
 
 @pytest.mark.asyncio
 async def test_router_timeouts_bedrock():
-    import openai, uuid
+    import uuid
+
+    import openai
 
     # Model list for OpenAI and Anthropic models
     _model_list = [