[CI] fix docs on `main` (#51539)
Signed-off-by: Harry Mellor <19981378+hmellor@users.noreply.github.com>
Harry Mellor committed
Aug 9, 2026 at 10:46 UTC
83ad767eed3be3ee7f2df63be693bfaca5c7c922
2 files changed
+2
-2
vllm/benchmarks/throughput.py
+1
-1
@@ -523,7 +523,7 @@ def save_to_pytorch_benchmark_format(
523
write_to_json(pt_file, pt_records)
524
525
526
-def _to_serve_args(args) -> argparse.Namespace:
526
+def _to_serve_args(args: argparse.Namespace) -> argparse.Namespace:
527
"""Translate throughput args into a namespace the shared get_samples reads.
528
529
``get_samples`` (used by ``bench serve``) expects ~45 attributes; the
vllm/multimodal/video.py
+1
-1
@@ -637,7 +637,7 @@ class TorchCodecVideoBackendMixin:
637
return batch.data.numpy(), list(frame_indices)
638
639
640
-def _pynvvc_frames_to_nhwc(frames):
640
+def _pynvvc_frames_to_nhwc(frames: torch.Tensor) -> torch.Tensor:
641
"""Return a stacked PyNvVideoCodec frame batch as contiguous NHWC.
642
643
PyNvVideoCodec's per-frame layout has varied across versions (HWC vs CHW),