From ea74f701db6c0dd4b2d954f5e79841101d0d8a5d Mon Sep 17 00:00:00 2001
From: zhrrr <43847754+izhuhaoran@users.noreply.github.com>
Date: Tue, 28 Apr 2026 15:33:49 +0800
Subject: [PATCH] Bugfix: fix SpecBench sample argument error (#40927)

Signed-off-by: zhuhaoran <zhuhaoran.zhr@alibaba-inc.com>
---
 vllm/benchmarks/datasets/datasets.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/vllm/benchmarks/datasets/datasets.py b/vllm/benchmarks/datasets/datasets.py
index 745b5ab2ff8..419275d2e6a 100644
--- a/vllm/benchmarks/datasets/datasets.py
+++ b/vllm/benchmarks/datasets/datasets.py
@@ -2365,6 +2365,7 @@ class SpecBench(CustomDataset):
             random.shuffle(self.data)
 
     def sample(
+        self,
         **kwargs,
     ) -> list[SampleRequest]:
         # leverage CustomDataset sample