mirror of
https://github.com/BerriAI/litellm.git
synced 2026-09-07 08:26:10 +00:00
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:
parent
5f79bf4906
commit
bae8816c35
1 changed files with 1 additions and 0 deletions
|
|
@ -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
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue