Skip to content

Commit 115e40f

Browse files
committed
nvidia vlm test: fix assert
1 parent b2ba40d commit 115e40f

1 file changed

Lines changed: 1 addition & 2 deletions

File tree

plugins/nvidia/tests/test_nvidia_vlm.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -86,8 +86,7 @@ async def test_with_video_frames(self, vlm: VLM, cat_frame: av.VideoFrame):
8686
)
8787

8888
assert final.text
89-
assert len(final.text) > 0
90-
assert "cat" in final.text.lower() or len(final.text.strip()) > 0
89+
assert "cat" in final.text.lower(), f"Expected 'cat' in response: {final.text}"
9190

9291
async def test_instruction_following(self, vlm):
9392
"""Test that system instructions are respected."""

0 commit comments

Comments
 (0)