From 7076e661b53cbe17d78bddb0315395d8c74f96f4 Mon Sep 17 00:00:00 2001 From: Aidan Do Date: Sun, 15 Dec 2024 12:05:52 +1100 Subject: [PATCH] Remove print statements in unit tests --- llama_stack/providers/tests/inference/groq/test_groq_utils.py | 1 - llama_stack/providers/tests/inference/test_text_inference.py | 1 - 2 files changed, 2 deletions(-) diff --git a/llama_stack/providers/tests/inference/groq/test_groq_utils.py b/llama_stack/providers/tests/inference/groq/test_groq_utils.py index b9191a1ba..db950511f 100644 --- a/llama_stack/providers/tests/inference/groq/test_groq_utils.py +++ b/llama_stack/providers/tests/inference/groq/test_groq_utils.py @@ -468,7 +468,6 @@ class TestConvertStreamChatCompletionResponse: iter = converted.__aiter__() chunk = await iter.__anext__() - print(chunk) assert chunk.event.event_type == ChatCompletionResponseEventType.start assert chunk.event.delta.content == ToolCall( call_id="tool_call_id", diff --git a/llama_stack/providers/tests/inference/test_text_inference.py b/llama_stack/providers/tests/inference/test_text_inference.py index d0c52c753..414d0261b 100644 --- a/llama_stack/providers/tests/inference/test_text_inference.py +++ b/llama_stack/providers/tests/inference/test_text_inference.py @@ -422,7 +422,6 @@ class TestInference: **common_params, ) ] - assert len(response) > 0 assert all( isinstance(chunk, ChatCompletionResponseStreamChunk) for chunk in response