Skip to content

Commit 6f17b10

Browse files
committed
style: ruff format --block_size argparse call
Multi-line per ruff-format; line 296 exceeded 120 char limit (E501). Signed-off-by: chenhany <chenhany@nvidia.com>
1 parent 618aceb commit 6f17b10

1 file changed

Lines changed: 10 additions & 1 deletion

File tree

examples/specdec_bench/run.py

Lines changed: 10 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -293,7 +293,16 @@ def run_simple(args):
293293
"--output_length", type=int, required=False, default=4096, help="Output length"
294294
)
295295
parser.add_argument("--draft_length", type=int, required=False, default=3, help="Draft length")
296-
parser.add_argument("--block_size", type=int, required=False, default=None, help="DFlash block size (num_speculative_tokens). Use instead of --draft_length for DFLASH: block_size = draft_length + 1.")
296+
parser.add_argument(
297+
"--block_size",
298+
type=int,
299+
required=False,
300+
default=None,
301+
help=(
302+
"DFlash block size (num_speculative_tokens). Use instead of --draft_length "
303+
"for DFLASH: block_size = draft_length + 1."
304+
),
305+
)
297306
parser.add_argument(
298307
"--tp_size", type=int, required=False, default=4, help="Tensor parallel size"
299308
)

0 commit comments

Comments
 (0)