From 4bba121f43efe53de665bfc2239a48a6ba4e8ad8 Mon Sep 17 00:00:00 2001 From: hiyouga Date: Sat, 23 Nov 2024 19:13:32 +0000 Subject: [PATCH] fix ci Former-commit-id: b52c38350d6092e8158484ca01da19ed016236fc --- tests/data/test_mm_plugin.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/data/test_mm_plugin.py b/tests/data/test_mm_plugin.py index 5ab91d2e..21d0ebad 100644 --- a/tests/data/test_mm_plugin.py +++ b/tests/data/test_mm_plugin.py @@ -61,7 +61,7 @@ INPUT_IDS = [0, 1, 2, 3, 4] LABELS = [0, 1, 2, 3, 4] -SEQLENS = [1024] +BATCH_IDS = [[1] * 1024] def _get_mm_inputs(processor: "ProcessorMixin") -> Dict[str, "torch.Tensor"]: @@ -105,7 +105,7 @@ def _check_plugin( expected_labels, ) _is_close( - plugin.get_mm_inputs(IMAGES, NO_VIDEOS, IMGLENS, NO_VIDLENS, SEQLENS, processor), + plugin.get_mm_inputs(IMAGES, NO_VIDEOS, IMGLENS, NO_VIDLENS, BATCH_IDS, processor), expected_mm_inputs, ) # test text_messages @@ -115,7 +115,7 @@ def _check_plugin( LABELS, ) _is_close( - plugin.get_mm_inputs(NO_IMAGES, NO_VIDEOS, NO_IMGLENS, NO_VIDLENS, SEQLENS, processor), + plugin.get_mm_inputs(NO_IMAGES, NO_VIDEOS, NO_IMGLENS, NO_VIDLENS, BATCH_IDS, processor), expected_no_mm_inputs, )