diff --git a/tests/litellm_utils_tests/test_bedrock_token_counter.py b/tests/litellm_utils_tests/test_bedrock_token_counter.py index b0e37914af..abc45b03d6 100644 --- a/tests/litellm_utils_tests/test_bedrock_token_counter.py +++ b/tests/litellm_utils_tests/test_bedrock_token_counter.py @@ -11,7 +11,7 @@ counting, the test will be skipped. import os import sys from typing import Any, Dict, List -from unittest.mock import AsyncMock, MagicMock, patch +from unittest.mock import patch import pytest