From a614c9f52544d9334bbd2dde99b4c22da0177e13 Mon Sep 17 00:00:00 2001
From: Ishaan Jaff <ishaanjaffer0324@gmail.com>
Date: Thu, 15 Aug 2024 21:55:27 -0700
Subject: [PATCH] fix sagemaker old used test

---
 litellm/tests/test_streaming.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/litellm/tests/test_streaming.py b/litellm/tests/test_streaming.py
index 025ea812009..3f423318792 100644
--- a/litellm/tests/test_streaming.py
+++ b/litellm/tests/test_streaming.py
@@ -1683,6 +1683,7 @@ def test_completion_bedrock_mistral_stream():
         pytest.fail(f"Error occurred: {e}")
 
 
+@pytest.mark.skip(reason="stopped using TokenIterator")
 def test_sagemaker_weird_response():
     """
     When the stream ends, flush any remaining holding chunks.