From dc17f63d0b4f101095dbd4cd55dc4155f9961d4c Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Thu, 23 Nov 2023 16:50:19 -0800 Subject: [PATCH] fix(router.py): fix linting errors --- litellm/router.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/litellm/router.py b/litellm/router.py index f93897e886..31a9f98bef 100644 --- a/litellm/router.py +++ b/litellm/router.py @@ -146,7 +146,7 @@ class Router: with concurrent.futures.ThreadPoolExecutor(max_workers=1) as executor: # Submit the function to the executor with a timeout future = executor.submit(self.function_with_fallbacks, **kwargs) - response = future.result(timeout=self.timeout) + response = future.result(timeout=self.timeout) # type: ignore return response except Exception as e: @@ -182,7 +182,7 @@ class Router: kwargs["num_retries"] = self.num_retries # Use asyncio.timeout to enforce the timeout - async with asyncio.timeout(self.timeout): + async with asyncio.timeout(self.timeout): # type: ignore response = await self.async_function_with_fallbacks(**kwargs) return response