fix(huggingface_restapi.py): return streamed response correctly

This commit is contained in:
Krrish Dholakia 2024-02-16 13:25:13 -08:00
parent a7d0f19e6c
commit 9c2cce56f7
2 changed files with 3 additions and 6 deletions

View file

@ -899,7 +899,7 @@ async def test_sagemaker_streaming_async():
pytest.fail(f"An exception occurred - {str(e)}")
asyncio.run(test_sagemaker_streaming_async())
# asyncio.run(test_sagemaker_streaming_async())
def test_completion_sagemaker_stream():