fix sagemaker old used test

This commit is contained in:
Ishaan Jaff 2024-08-15 21:55:27 -07:00
parent eb9e5af1a3
commit a614c9f525

View file

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