fix typo/lint

Signed-off-by: Linkun Chen <[email protected]>

benchmarks/benchmark_throughput.py

@@ -44,10 +44,12 @@ def _get_prompt_for_image_model(question: str, *, model: str) -> str:
 
     Args:
         question: The input question text to wrap with special tokens
-        model: The name of the model being used, to determine which special tokens to add
+        model: The name of the model being used, to determine which special
+            tokens to add
 
     Returns:
-        The formatted prompt string with appropriate special tokens for the model
+        The formatted prompt string with appropriate special tokens for the
+            model
 
     Raises:
         ValueError: If an unsupported model name is provided
@@ -105,7 +107,7 @@ def sample_requests(tokenizer: PreTrainedTokenizerBase,
         completion_token_ids = tokenizer(completion).input_ids
         prompt_len = len(prompt_token_ids)
         output_len = len(completion_token_ids
-                        ) if fixed_output_len is None else fixed_output_len
+                         ) if fixed_output_len is None else fixed_output_len
         if prompt_len < 4 or output_len < 4:
             # Prune too short sequences.
             continue