Unverified Commit 06724683 authored by DarkSharpness's avatar DarkSharpness Committed by GitHub
Browse files

[Minor] fix lint in main (#10128)

parent 617aa2b2
...@@ -85,7 +85,10 @@ class BenchArgs: ...@@ -85,7 +85,10 @@ class BenchArgs:
) )
parser.add_argument("--profile-by-stage", action="store_true") parser.add_argument("--profile-by-stage", action="store_true")
parser.add_argument( parser.add_argument(
"--dataset-path", type=str, default=BenchArgs.dataset_path, help="Path to the dataset." "--dataset-path",
type=str,
default=BenchArgs.dataset_path,
help="Path to the dataset.",
) )
@classmethod @classmethod
...@@ -350,7 +353,7 @@ def run_benchmark(server_args: ServerArgs, bench_args: BenchArgs): ...@@ -350,7 +353,7 @@ def run_benchmark(server_args: ServerArgs, bench_args: BenchArgs):
run_name="", run_name="",
result_filename="", result_filename="",
tokenizer=tokenizer, tokenizer=tokenizer,
dataset_path=bench_args.dataset_path dataset_path=bench_args.dataset_path,
) )
print("=" * 8 + " Warmup End " + "=" * 8 + "\n") print("=" * 8 + " Warmup End " + "=" * 8 + "\n")
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment