Update test_cli_vlm.py

sbalandi · web-flow · commit e11ba064b25e · 2025-11-28T19:52:36.000Z
diff --git a/tools/who_what_benchmark/tests/test_cli_vlm.py b/tools/who_what_benchmark/tests/test_cli_vlm.py
@@ -115,11 +115,11 @@ def test_vlm_nanollava(model_id, model_type, optimum_threshold, genai_threshold,
 
 
 @pytest.mark.parametrize(
-    ("model_id", "model_type", "threshold"),
+    ("model_id", "model_type"),
     [
         ("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):
+def test_vlm_video(model_id, model_type, tmp_path):
     run_test(model_id, model_type, None, None, tmp_path)

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