fix linting errors

BerriAI · Dec 20, 2024 · 2c36f25 · 2c36f25
1 parent 834067c
commit 2c36f25
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/litellm/rerank_api/main.py b/litellm/rerank_api/main.py
@@ -205,11 +205,9 @@ def rerank(  # noqa: PLR0915
             )
         elif _custom_llm_provider == "infinity":
             # Implement Infinity rerank logic
-            api_key: Optional[str] = (
-                dynamic_api_key or optional_params.api_key or litellm.api_key
-            )
+            api_key = dynamic_api_key or optional_params.api_key or litellm.api_key
 
-            api_base: Optional[str] = (
+            api_base = (
                 dynamic_api_base
                 or optional_params.api_base
                 or litellm.api_base