From 84513c02548b549d7beb91be85982ce7cb77c961 Mon Sep 17 00:00:00 2001
From: Krrish Dholakia <krrishdholakia@gmail.com>
Date: Tue, 30 Jul 2024 15:34:27 -0700
Subject: [PATCH] test(test_streaming.py): move to mock implementation for
 sagemaker streaming tests

---
 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 ed88f818cc0..cebac19ea18 100644
--- a/litellm/tests/test_streaming.py
+++ b/litellm/tests/test_streaming.py
@@ -1766,6 +1766,7 @@ async def test_sagemaker_streaming_async():
 # asyncio.run(test_sagemaker_streaming_async())
 
 
+@pytest.mark.skip(reason="costly sagemaker deployment. Move to mock implementation")
 def test_completion_sagemaker_stream():
     try:
         response = completion(