Skip to content

Commit

Permalink
Cleanup
Browse files Browse the repository at this point in the history
Signed-off-by: DarkLight1337 <[email protected]>
  • Loading branch information
DarkLight1337 committed Dec 23, 2024
1 parent 5fee280 commit 6182fd6
Showing 1 changed file with 2 additions and 7 deletions.
9 changes: 2 additions & 7 deletions tests/multimodal/test_processing.py
Original file line number Diff line number Diff line change
Expand Up @@ -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]())
Expand All @@ -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})"

0 comments on commit 6182fd6

Please sign in to comment.