Bugfix: fix SpecBench sample argument error (#40927)

Signed-off-by: zhuhaoran <zhuhaoran.zhr@alibaba-inc.com>
2026-04-28 00:33:49 -07:00
parent a8208e6a81
commit ea74f701db
1 changed files with 1 additions and 0 deletions
@@ -2365,6 +2365,7 @@ class SpecBench(CustomDataset):
            random.shuffle(self.data)

    def sample(
+        self,
        **kwargs,
    ) -> list[SampleRequest]:
        # leverage CustomDataset sample