diff --git a/tests/test_litellm/proxy/test_health_check_max_tokens.py b/tests/test_litellm/proxy/test_health_check_max_tokens.py index 6fc87e3665d..85e3aa22596 100644 --- a/tests/test_litellm/proxy/test_health_check_max_tokens.py +++ b/tests/test_litellm/proxy/test_health_check_max_tokens.py @@ -39,10 +39,12 @@ async def test_update_litellm_params_max_tokens_custom(): @pytest.mark.asyncio -async def test_update_litellm_params_max_tokens_wildcard(): +async def test_update_litellm_params_max_tokens_wildcard(monkeypatch): """ Test that max_tokens is not set for wildcard models. """ + monkeypatch.setattr(hc_module, "BACKGROUND_HEALTH_CHECK_MAX_TOKENS", None) + monkeypatch.setattr(hc_module, "BACKGROUND_HEALTH_CHECK_MAX_TOKENS_REASONING", None) model_info = {} litellm_params = {"model": "openai/*"}