Update tests/test_litellm/litellm_core_utils/test_token_counter.py

Co-authored-by: greptile-apps[bot] <165735046+greptile-apps[bot]@users.noreply.github.com>
This commit is contained in:
jquinter 2026-02-15 20:27:29 -03:00 committed by GitHub
parent 5f79bf4906
commit bae8816c35
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -517,6 +517,7 @@ class TestTokenizerSelection(unittest.TestCase):
def setUp(self):
"""Clear cache before each test method."""
_select_tokenizer_helper.cache_clear()
@patch("litellm.utils.Tokenizer.from_pretrained")
def test_llama3_tokenizer_api_failure(self, mock_from_pretrained):
# Setup mock to raise an error