Update test_cli_vlm.py

sbalandi · web-flow · commit 23f0ef95b752 · 2025-11-28T18:55:03.000Z
diff --git a/tools/who_what_benchmark/tests/test_cli_vlm.py b/tools/who_what_benchmark/tests/test_cli_vlm.py
@@ -117,9 +117,9 @@ def test_vlm_nanollava(model_id, model_type, optimum_threshold, genai_threshold,
 @pytest.mark.parametrize(
     ("model_id", "model_type", "threshold"),
     [
-        ("katuni4ka/tiny-random-qwen2vl", "visual-text", 0.8),
-        ("katuni4ka/tiny-random-llava-next-video", "visual-text", 0.8),
+        ("katuni4ka/tiny-random-qwen2vl", "visual-text"),
+        ("katuni4ka/tiny-random-llava-next-video", "visual-text"),
     ],
 )
 def test_vlm_video(model_id, model_type, threshold, tmp_path):
-    run_test(model_id, model_type, threshold, threshold, tmp_path)
+    run_test(model_id, model_type, None, None, tmp_path)

Original file line number	Diff line number	Diff line change
`@@ -117,9 +117,9 @@ def test_vlm_nanollava(model_id, model_type, optimum_threshold, genai_threshold,`
`117`	`117`	`@pytest.mark.parametrize(`
`118`	`118`	`("model_id", "model_type", "threshold"),`
`119`	`119`	`[`
`120`		`- ("katuni4ka/tiny-random-qwen2vl", "visual-text", 0.8),`
`121`		`- ("katuni4ka/tiny-random-llava-next-video", "visual-text", 0.8),`
	`120`	`+ ("katuni4ka/tiny-random-qwen2vl", "visual-text"),`
	`121`	`+ ("katuni4ka/tiny-random-llava-next-video", "visual-text"),`
`122`	`122`	`],`
`123`	`123`	`)`
`124`	`124`	`def test_vlm_video(model_id, model_type, threshold, tmp_path):`
`125`		`- run_test(model_id, model_type, threshold, threshold, tmp_path)`
	`125`	`+ run_test(model_id, model_type, None, None, tmp_path)`