diff --git a/tests/test_litellm/caching/test_s3_cache.py b/tests/test_litellm/caching/test_s3_cache.py index 85c62cc462..7fbb7f316c 100644 --- a/tests/test_litellm/caching/test_s3_cache.py +++ b/tests/test_litellm/caching/test_s3_cache.py @@ -1,4 +1,3 @@ -from litellm.caching.s3_cache import S3Cache import os import sys from unittest.mock import MagicMock, patch @@ -12,6 +11,8 @@ sys.path.insert( 0, os.path.abspath("../../..") ) # Adds the parent directory to the system path +from litellm.caching.s3_cache import S3Cache + @pytest.fixture def mock_s3_dependencies():