diff --git a/tests/test_litellm/litellm_core_utils/test_token_counter.py b/tests/test_litellm/litellm_core_utils/test_token_counter.py index a562e72466..0ce16de39f 100644 --- a/tests/test_litellm/litellm_core_utils/test_token_counter.py +++ b/tests/test_litellm/litellm_core_utils/test_token_counter.py @@ -582,14 +582,15 @@ class TestTokenizerSelection(unittest.TestCase): @patch("litellm.utils._return_huggingface_tokenizer") def test_disable_hf_tokenizer_download(self, mock_return_huggingface_tokenizer): - # Use pytest.MonkeyPatch() directly instead of fixture monkeypatch = pytest.MonkeyPatch() monkeypatch.setattr(litellm, "disable_hf_tokenizer_download", True) - - result = _select_tokenizer_helper("grok-32r22r") - mock_return_huggingface_tokenizer.assert_not_called() - assert result["type"] == "openai_tokenizer" - assert result["tokenizer"] == encoding + try: + result = _select_tokenizer_helper("grok-32r22r") + mock_return_huggingface_tokenizer.assert_not_called() + assert result["type"] == "openai_tokenizer" + assert result["tokenizer"] == encoding + finally: + monkeypatch.undo() @pytest.mark.parametrize(