From 677e3f5d5040db80087fd5693cd3434d1adcde83 Mon Sep 17 00:00:00 2001 From: Cyrus Leung Date: Wed, 30 Sep 2026 21:45:28 +0800 Subject: [PATCH] [CI/Build] Fix pre-commit (#59426) --- tests/multimodal/test_cache.py | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/tests/multimodal/test_cache.py b/tests/multimodal/test_cache.py index 0416734c24..da6b0369b4 100644 --- a/tests/multimodal/test_cache.py +++ b/tests/multimodal/test_cache.py @@ -2,6 +2,7 @@ # SPDX-FileCopyrightText: Copyright contributors to the vLLM project import multiprocessing as mp from types import SimpleNamespace +from typing import cast from unittest.mock import Mock import numpy as np @@ -796,6 +797,7 @@ def test_shm_receiver_requires_sender_handle_for_current_cache_key(): try: sender_item, _ = sender_cache.get_and_update_item((item, []), mm_hash) + assert sender_item is not None valid_feature = MultiModalFeatureSpec( data=sender_item, @@ -897,10 +899,12 @@ def test_shm_sender_validates_client_supplied_handles(): handle, _ = sender_cache.get_and_update_item( (MultiModalKwargsItem.dummy(1024), []), mm_hash ) + assert handle is not None + sender_cache.validate_input_item(handle, mm_hash) sender_cache.validate_input_item(MultiModalKwargsItem.dummy(16), mm_hash) - signature = handle["signature"].data + signature = cast(list[int], handle["signature"].data) forged = [ _with_elem(handle, "signature", [signature[0] ^ 1, *signature[1:]]), _with_elem(handle, "signature", [signature]), @@ -934,6 +938,8 @@ def test_shm_receiver_rejects_out_of_range_handle_fields(): handle, _ = sender_cache.get_and_update_item( (MultiModalKwargsItem.dummy(1024), []), mm_hash ) + assert handle is not None + for key, value in [ ("address", -1), ("address", 2**64), @@ -967,6 +973,8 @@ def test_shm_handle_issued_before_cache_clear(): try: stale_item, _ = sender_cache.get_and_update_item((item, []), mm_hash) + assert stale_item is not None + sender_cache.clear_cache() # A request still draining after the clear resolves to its own data.