[data] fix pixtral plugin (#7505)

* preserve `image_sizes` * add comments
2025-03-27 17:06:40 +08:00
parent b578a7d5b6
commit 8da1d2fa71
2 changed files with 10 additions and 3 deletions
--- a/tests/data/test_mm_plugin.py
+++ b/tests/data/test_mm_plugin.py
@@ -232,7 +232,6 @@ def test_pixtral_plugin():
        for message in MM_MESSAGES
    ]
    check_inputs["expected_mm_inputs"] = _get_mm_inputs(tokenizer_module["processor"])
-    check_inputs["expected_mm_inputs"].pop("image_sizes")
    check_inputs["expected_mm_inputs"]["pixel_values"] = check_inputs["expected_mm_inputs"]["pixel_values"][0]
    _check_plugin(**check_inputs)