diff --git a/tests/python_tests/test_vlm_api.py b/tests/python_tests/test_vlm_api.py index 767e4b754d..4b43dd99c6 100644 --- a/tests/python_tests/test_vlm_api.py +++ b/tests/python_tests/test_vlm_api.py @@ -65,7 +65,7 @@ def streamer(word: str) -> bool: for prompt in prompts[1:]: result_from_streamer = [] res = pipe.generate(prompt, generation_config=get_greedy(), streamer=streamer) - assert res.texts[0] == ''.join(result_from_streamer) + assert res.texts[0] == ''.join(result_from_streamer) pipe.finish_chat()