diff --git a/benchmarks/benchmark_throughput.py b/benchmarks/benchmark_throughput.py
index 3e3e4272f2ba4..a1080ebc59e76 100644
--- a/benchmarks/benchmark_throughput.py
+++ b/benchmarks/benchmark_throughput.py
@@ -394,6 +394,7 @@ def validate_args(args):
     if args.backend not in valid_backends:
         raise ValueError(f"Unsupported backend: {args.backend}")
 
+    # === Dataset Validation ===
     validate_dataset(args)
 
     # === LoRA Settings ===