diff --git a/vllm/multimodal/inputs.py b/vllm/multimodal/inputs.py index 151cf56db1e3b..ebaaffe7ea4a9 100644 --- a/vllm/multimodal/inputs.py +++ b/vllm/multimodal/inputs.py @@ -127,7 +127,7 @@ def nested_tensors_equal(a: NestedTensors, b: NestedTensors) -> bool: return a == b -BatchedTensorInputs: TypeAlias = dict[str, NestedTensors] +BatchedTensorInputs: TypeAlias = Mapping[str, NestedTensors] """ A dictionary containing nested tensors which have been batched via :meth:`MultiModalKwargs.batch`.