diff --git a/litellm/tests/test_streaming.py b/litellm/tests/test_streaming.py index f3082c93f4e..f69dce9a3d0 100644 --- a/litellm/tests/test_streaming.py +++ b/litellm/tests/test_streaming.py @@ -6,7 +6,6 @@ import traceback import time, pytest from pydantic import BaseModel from typing import Tuple -from unittest.mock import patch, MagicMock sys.path.insert( 0, os.path.abspath("../..")