From 38988f030ac74c8702473a76a43a070fe706635a Mon Sep 17 00:00:00 2001
From: Krrish Dholakia <krrishdholakia@gmail.com>
Date: Mon, 13 May 2024 18:06:10 -0700
Subject: [PATCH] fix(router.py): fix typing

---
 litellm/router.py                | 2 +-
 litellm/tests/test_azure_perf.py | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/litellm/router.py b/litellm/router.py
index fe8709294..b4603c6d0 100644
--- a/litellm/router.py
+++ b/litellm/router.py
@@ -63,7 +63,7 @@ class Router:
 
     def __init__(
         self,
-        model_list: Optional[List[DeploymentTypedDict]] = None,
+        model_list: Optional[List[Union[DeploymentTypedDict, Dict]]] = None,
         ## CACHING ##
         redis_url: Optional[str] = None,
         redis_host: Optional[str] = None,
diff --git a/litellm/tests/test_azure_perf.py b/litellm/tests/test_azure_perf.py
index 9654f1273..8afc59f92 100644
--- a/litellm/tests/test_azure_perf.py
+++ b/litellm/tests/test_azure_perf.py
@@ -26,7 +26,7 @@ model_list = [
     }
 ]
 
-router = litellm.Router(model_list=model_list)
+router = litellm.Router(model_list=model_list)  # type: ignore
 
 
 async def _openai_completion():