From 6182fd64f735d50ec997bcb2a167f54092bb9378 Mon Sep 17 00:00:00 2001 From: DarkLight1337 Date: Mon, 23 Dec 2024 09:24:50 +0000 Subject: [PATCH] Cleanup Signed-off-by: DarkLight1337 --- tests/multimodal/test_processing.py | 9 ++------- 1 file changed, 2 insertions(+), 7 deletions(-) diff --git a/tests/multimodal/test_processing.py b/tests/multimodal/test_processing.py index 4a370b73a7add..033c9d838b755 100644 --- a/tests/multimodal/test_processing.py +++ b/tests/multimodal/test_processing.py @@ -587,9 +587,7 @@ def test_processing_cache_correctness( "audio": 3, } - baseline_results = [] - cached_results = [] - for _ in range(num_batches): + for batch_idx in range(num_batches): mm_data = { k: [(input_to_hit[k] if rng.rand() < hit_rate else input_factory[k]()) @@ -613,13 +611,10 @@ def test_processing_cache_correctness( mm_data=mm_data, hf_processor_mm_kwargs={}, ) - baseline_results.append(baseline_result) - cached_result = cached_processor.apply( prompt, mm_data=mm_data, hf_processor_mm_kwargs={}, ) - cached_results.append(cached_result) - assert baseline_results == cached_results + assert baseline_result == cached_result, f"Failed (idx={batch_idx})"