diff --git a/vllm/benchmarks/throughput.py b/vllm/benchmarks/throughput.py index 894334a7cf..e8a0ccb3a0 100644 --- a/vllm/benchmarks/throughput.py +++ b/vllm/benchmarks/throughput.py @@ -523,7 +523,7 @@ def save_to_pytorch_benchmark_format( write_to_json(pt_file, pt_records) -def _to_serve_args(args) -> argparse.Namespace: +def _to_serve_args(args: argparse.Namespace) -> argparse.Namespace: """Translate throughput args into a namespace the shared get_samples reads. ``get_samples`` (used by ``bench serve``) expects ~45 attributes; the diff --git a/vllm/multimodal/video.py b/vllm/multimodal/video.py index 05c6daf3fc..068071dfe0 100644 --- a/vllm/multimodal/video.py +++ b/vllm/multimodal/video.py @@ -637,7 +637,7 @@ class TorchCodecVideoBackendMixin: return batch.data.numpy(), list(frame_indices) -def _pynvvc_frames_to_nhwc(frames): +def _pynvvc_frames_to_nhwc(frames: torch.Tensor) -> torch.Tensor: """Return a stacked PyNvVideoCodec frame batch as contiguous NHWC. PyNvVideoCodec's per-frame layout has varied across versions (HWC vs CHW),