Skip to content

Commit 6466d68

Browse files
committed
Fix format
Signed-off-by: Abhishek <[email protected]>
1 parent 685261d commit 6466d68

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

aiu_fms_testing_utils/scripts/drive_paged_programs.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -289,7 +289,9 @@ def __prepare_inputs(batch_size, seq_length, tokenizer, enforce_sizes=[], seed=0
289289
prompt_list.append(encoded)
290290

291291
if not prompt_list:
292-
raise ValueError(f"No valid prompt sample exists in dataset for input shape (Batch Size={batch_size}, Seq Length={seq_length})")
292+
raise ValueError(
293+
f"No valid prompt sample exists in dataset for input shape (Batch Size={batch_size}, Seq Length={seq_length})"
294+
)
293295
if len(prompt_list) < batch_size:
294296
dprint(
295297
f"You requested {batch_size} prompts but we were only able to get {len(prompt_list)} valid prompts. We will be repeating the first prompt."

0 commit comments

Comments
 (0)