From 687b2e6300f51bd0ccb48ed516127a2c0a57a993 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Tue, 11 Mar 2025 18:13:27 -0700 Subject: [PATCH] test: fix test --- tests/litellm/llms/azure/test_azure_common_utils.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/tests/litellm/llms/azure/test_azure_common_utils.py b/tests/litellm/llms/azure/test_azure_common_utils.py index f4322a5a9c0..b74acc518f0 100644 --- a/tests/litellm/llms/azure/test_azure_common_utils.py +++ b/tests/litellm/llms/azure/test_azure_common_utils.py @@ -44,7 +44,9 @@ def setup_mocks(): mock_oidc_token.return_value = "mock-oidc-token" mock_token_provider.return_value = lambda: "mock-default-token" - mock_select_url.side_effect = lambda params: params + mock_select_url.side_effect = ( + lambda azure_client_params, **kwargs: azure_client_params + ) yield { "entrata_token": mock_entrata_token,