From 83ad767eed3be3ee7f2df63be693bfaca5c7c922 Mon Sep 17 00:00:00 2001 From: Harry Mellor <19981378+hmellor@users.noreply.github.com> Date: Sun, 9 Aug 2026 10:46:12 +0100 Subject: [PATCH] [CI] fix docs on `main` (#51539) Signed-off-by: Harry Mellor <19981378+hmellor@users.noreply.github.com> --- vllm/benchmarks/throughput.py | 2 +- vllm/multimodal/video.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) 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),