test(test_streaming.py): move to mock implementation for sagemaker streaming tests

This commit is contained in:
Krrish Dholakia 2024-07-30 15:34:27 -07:00
parent e1cbb397fe
commit 84513c0254

View file

@ -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(